如何为数据框中的每一列创建一个csv文件?

时间:2020-09-12 19:51:08

标签: python pandas csv split

我在VSCode中使用Python。我读取了一个CSV文件,并在一个数据框中将其转换为股票市场的收盘价。第一列是日期列,其他各是股票代码的收盘价。我想将此具有多个列的数据框拆分为单独的.csv文件。每个新的.csv文件将按照每个股票代码列标题的名称来命名。

下面是数据框的链接作为示例(原始文件具有500个coluns和10,000行):

因此,每个新的.csv文件将被命名为"TAEE11.csv",依此类推,在该列下包含各自的值,同时也为每个文件保留“ Date”列。我想将这些新创建的.csv文件保存到新文件夹(例如,当您在VSCode中使用Python时,可以使用*.to_csv将这些新的.csv文件保存到新文件夹)。

1 个答案:

答案 0 :(得分:2)

  • 因为您已经有一个数据框
  • 将数据框的索引设置为'Date'
  • 遍历每列并将其保存到csv。
    • 使用df[col]
    • 为每一列选择数据
    • csv文件名将为f'{col}.csv',其中col是列名。
  • 要在VSCode中工作时指定特定的文件保存位置,请参阅此SO: Answer
    • df.to_csv(os.getcwd()+'\\file.csv')进入AppData文件夹
    • 当前工作目录不一定是您想要保存文件的位置。
    • 例如,指定所需的完整保存路径,'c:/Users/<<user_name>>/Documents/{col}.csv'
import pandas as pd

# set Date as the index
df.set_index('Date', inplace=True)

# iterate through each column and save it
for col in df.columns:
    df[col].to_csv(f'c:/Users/<<user_name>>/Documents/{col}.csv', index=True)