从for循环写入多个CSV文件

时间:2019-10-30 03:32:51

标签: python csv csv-write-stream

我有一个for loop,它得到股票代码,另一个for循环得到股票数据。现在,我正在尝试创建csv文件,该文件将从第一个for循环中获取股票名称数据,并将股票数据添加到CSV文件中。

stock_data = []
with open('Nifty 50 Scrapped data.csv') as csvfile:
    stockticker_data = csv.reader(csvfile, delimiter=' ')
    for row in stockticker_data:
        # print(row)
        all_data = []
        for ticker in row:
            stock_data.append(web.get_data_yahoo(ticker, '1/1/2018', '1/1/2019'))
            with open(ticker, 'w') as f:
                f = open(ticker,'w')
                f.write(stock_data)
                f.close()

我遇到以下错误:

    f.write(stock_data)
TypeError: write() argument must be str, not list

print(stock_data)的输出:

[                   High          Low  ...     Volume    Adj Close
Date                                  ...                        
2018-01-01  1165.000000  1138.099976  ...   591349.0  1127.366211
2018-01-02  1150.000000  1134.050049  ...   516171.0  1126.479004
2018-01-03  1149.000000  1135.300049  ...   593809.0  1125.641235
2018-01-04  1178.000000  1145.900024  ...   729965.0  1155.361816
2018-01-05  1192.000000  1167.449951  ...  1151320.0  1168.373901
...                 ...          ...  ...        ...          ...
2018-12-27  1384.750000  1354.300049  ...  2174090.0  1362.613403
2018-12-28  1383.000000  1359.000000  ...  1705033.0  1356.160278
2018-12-31  1378.000000  1367.300049  ...   698593.0  1363.159546
2019-01-01  1379.699951  1358.599976  ...   664707.0  1361.670410
2019-01-02  1386.849976  1361.599976  ...  1233780.0  1373.335693

[248 rows x 6 columns]]
[                   High          Low  ...     Volume    Adj Close
Date                                  ...                        
2018-01-01  1165.000000  1138.099976  ...   591349.0  1127.366211
2018-01-02  1150.000000  1134.050049  ...   516171.0  1126.479004
2018-01-03  1149.000000  1135.300049  ...   593809.0  1125.641235
2018-01-04  1178.000000  1145.900024  ...   729965.0  1155.361816
2018-01-05  1192.000000  1167.449951  ...  1151320.0  1168.373901
...                 ...          ...  ...        ...          ...
2018-12-27  1384.750000  1354.300049  ...  2174090.0  1362.613403
2018-12-28  1383.000000  1359.000000  ...  1705033.0  1356.160278
2018-12-31  1378.000000  1367.300049  ...   698593.0  1363.159546
2019-01-01  1379.699951  1358.599976  ...   664707.0  1361.670410
2019-01-02  1386.849976  1361.599976  ...  1233780.0  1373.335693

1 个答案:

答案 0 :(得分:2)

此错误告诉您stock_datalist,而write()方法期望使用str。知道从web.get_data_yahoo()返回的数据是pd.DataFrame,该如何解决?我们可以像这样使用pd.to_csv来做到这一点:

stock_data = []
with open('Nifty 50 Scrapped data.csv') as csvfile:
    stockticker_data = csv.reader(csvfile, delimiter=' ')
    for row in stockticker_data:
        # print(row)
        all_data = []
        for ticker in row:
            stock_data.append(web.get_data_yahoo(ticker, '1/1/2018', '1/1/2019'))
            for df in stock_data:
                df.to_csv(ticker, header=None, index=None, sep=' ', mode='a')