python 如何将txt文件中的列名导入Pandas数据框?

hlswsv35  于 2023-02-07  发布在  Python
关注(0)|答案(1)|浏览(120)

我有一个txt文件下面不同的列名,我想提取所有的列名没有数据类型等从表到Pandas Dataframe 。

create table ad.TrackData
(
    track_id         int unsigned auto_increment primary key,
    ads_id           int                  null,
    ads_name         varchar(45)          null,
    play_time        timestamp            null,
    package_id       int                  null,
    package_name     varchar(45)          null,
    company_id       int                  null,
    company_name     varchar(45)          null,
    click_time       timestamp            null,
    demographic      varchar(300)         null,
    status           tinyint(1) default 0 null
);

我不知道我应该如何做这件事,如果有人能教我一些方法来执行这一点,我将非常感激。

s8vozzvw

s8vozzvw1#

首先,你必须将文本保存到file.txt,然后运行下面的python脚本:

with open('file.txt', 'r') as f:
  txt = f.read()
  lines = txt.split('\n')
  columnNames = []
  for line in lines:
    if line.startswith('    '):
      columnNames.append(line.split('    ')[1])
  print(columnNames)

输出必须为:

['track_id', 'ads_id', 'ads_name', 'play_time', 'package_id', 'package_name', 'company_id', 'company_name', 'click_time', 'demographic', 'status']

相关问题