Python寫入列表到csv

如何將打印語句中的結果寫入csv文件，每次循環運行結果時都要添加到下一行。結果包含一個列表。Python寫入列表到csv

def func_high(i): 
    swing_low = data_file.iloc[i].Low  
    date = data_file.iloc[i].Date 

    while i < len(data_file): 
     i = i + 1 

     try : 
      swing_high = data_file.iloc[i].High 
      if (swing_high < data_file.iloc[i + 1].High) and (data_file.iloc[i + 1].High > data_file.iloc[i + 2].High): 
       date_diff = data_file.iloc[i + 1].Date - date 
       price_diff = data_file.iloc[i + 1].High - swing_low 
       **print (price_diff/date_diff.days, date, swing_low , data_file.iloc[i + 1].Date, data_file.iloc[i + 1].High, date_diff)** 

    except IndexError: 
     pass 
    continue

來源

2016-08-11 user1779646

您可以將print更改爲yield，並使用csv.writer（我對您的功能進行了一些其他編輯，以便我可以更輕鬆地執行此邏輯）。

import csv 
i = 0 

def func_high(i): 
    swing_low = data_file.iloc[i].Low  
    date = data_file.iloc[i].Date 
    while i < (len(data_file) - 3): 
     i += 1 

     swing_high = data_file.iloc[i].High 
     curr_date = data_file.iloc[i + 1].Date 
     curr_high = data_file.iloc[i + 1].High 
     next_high = data_file.iloc[i + 2].High 

     if (swing_high < curr_high) and (curr_high > next_high): 
      date_diff = curr_date - date 
      price_diff = curr_high - swing_low 
      avg_change = price_diff/date_diff.days 
      yield [avg_change, date, swing_low, curr_date, curr_high, date_diff] 


with open('my_csv.csv', 'w') as out_f: 
    writer = csv.writer(out_f, lineterminator='\n') 
    writer.writerows(func_high(i))

如果你想要去的大熊貓的路線，你可以做使用上述發電機以下。

pd.DataFrame(list(func_high(i)).to_csv('my_csv.csv', index=False, header=False)

來源

2016-08-11 18:09:58 JaminSore

謝謝，我喜歡你編輯的。 – user1779646

如果結果是Series or DataFrame你應該只使用to_csv方法的大熊貓。

來源

2016-08-11 15:08:52

所以，我會創建一個數據框並追加結果？ – user1779646

您似乎可以通過簡單地生成一個序列/數據框的方式獲取這些信息。我對熊貓還不太瞭解，但還不知道該怎麼做。 –

這不是一個答案，應該是一個評論 – Merlin

最簡單的就是使用熊貓。

import pandas as pd 

mylist = [1, 2, 3, 4] 

df = pd.Series(mylist) 

df.to_csv('filename.csv')

您必須將列表轉換爲序列/數據框。

來源

2016-08-11 15:39:53 tadamhicks

如果您使用的是熊貓的數據幀，請參閱該交易的其他的答案之一。如果沒有，使用CSV編寫器也可以。但是，爲了完整起見，我會指出有時最簡單的方法就是直接寫數據。不知道什麼進入您的示例代碼中的數據結構，我提供了一個簡單的示例程序：

import sys 

def write_csv_row(f, row_data): 
    out_str = ','.join([str(e) for e in row_data]) 
    f.write(out_str + '\n') 

def main(): 
    fname = 'test.dat' 
    data = [["Column 1", "Column 2"], [1,1], [2,2]] 
    with open(fname, 'w') as out_file: 
     for row in data: 
      write_csv_row(out_file, row) 

if __name__ == '__main__': 
    sys.exit(main())

來源

2016-08-12 15:49:17 GreenMatt

Python寫入列表到csv

回答

相關問題