python - 打开 ZIP 文件，扫描其中的 CSV 文件并将某些内容传输到另一个 CSV 文件

Question

以下代码查看 ZIP 文件夹中的 CSV 文件，然后使用索引设置的概念，尝试提取每个 CSV 文件的“SYMBOL”列中包含条目“HINDUNILVR”的行(此代码段在另一个循环内)，用于一系列日期(阅读 ZIP 文件)。但是，代码中存在一些问题，因为 C: 中存在的“stock.csv”文件中没有输出。运行此代码后为空。

预期输出文件说明:

一个名为“stock.csv”的 CSV 文件，其中第一列为 SYMBOL，HINDUNILVR 作为该列每个单元格中的条目，随后是每天与股票相关的许多价格和详细信息，在同一行。

Python 代码:

if(int(str(sdate.day))<10):

     df1 = pd.read_csv(zf.open("cm" + "0" + str(sdate.day) + MonthCode(sdate.month) + str(sdate.year) + "bhav.csv"))

   else:

     df1 = pd.read_csv(zf.open("cm" + str(sdate.day) + MonthCode(sdate.month) + str(sdate.year) + "bhav.csv"))

   df2 = df1.set_index("SYMBOL", drop = False)

   saved_row = df2.loc["HINDUNILVR",:]

      with open_file('C:\stock.csv','a') as f:
        saved_row.to_csv(f,header=False)

Answer 1

据我了解，您只是希望将 SYMBOL='HINDUNILVR' 所在的所有行复制到一个新文件中。下面的示例代码读取同一个文件并写回同一个文件。您可以根据需要更改文件名:

import pandas as pd
import os
df1 = pd.read_csv("Sample.csv")
df2 = df1[df1["SYMBOL"]=="HINDUNILVR"]
with open('Sample.csv','a+') as f:
    df2.to_csv(f,header=False,index=False)

或

import pandas as pd
import os
df1 = pd.read_csv("Sample.csv")
df2 = df1[df1["SYMBOL"]=="HINDUNILVR"]
df2.to_csv("Sample.csv",header=False,mode='a+',index=False)

或更新您的代码:

 if(int(str(sdate.day))<10):
    df1 = pd.read_csv(zf.open("cm" + "0" + str(sdate.day) + MonthCode(sdate.month) + str(sdate.year) + "bhav.csv"))
else:
    df1 = pd.read_csv(zf.open("cm" + str(sdate.day) + MonthCode(sdate.month) + str(sdate.year) + "bhav.csv"))
df2 = df1[df1["SYMBOL"]=="HINDUNILVR"]
df2.to_csv("C:\stock.csv",mode='a+',header=False,index=False)

或

    if(int(str(sdate.day))<10):
    df1 = pd.read_csv(zf.open("cm" + "0" + str(sdate.day) + MonthCode(sdate.month) + str(sdate.year) + "bhav.csv"))
else:
    df1 = pd.read_csv(zf.open("cm" + str(sdate.day) + MonthCode(sdate.month) + str(sdate.year) + "bhav.csv"))
df2 = df1[df1["SYMBOL"]=="HINDUNILVR"]
with open('C:\stock.csv','a+') as f:
    df2.to_csv(f,header=False,index=False)