python - 将值直接循环到 csv

Question

为简单起见，假设我有以下内容:

for i in range(0 ,100):
   value1 = i*0.01  
   value2 = i*2.75 

如何使用 pandas 将 value1 和 value2 写入 csv 文件，以便结果如下所示:

value1        value2 
0               0
0.01            2.75
0.02            5.25
.....            ...

我一直在搜索，但可以找到任何使用简单值作为 csv 文件输入的示例。知道我该怎么做!

Answer 1

如果您致力于使用 pandas，则可以使用字典以及 DataFrame 的 .from_dict() 和 to_csv() 方法:

import pandas as pd

data = {'value1': [i*0.01 for i in range(100)],
        'value2': [i*2.75 for i in range(100)]}

pd.DataFrame.from_dict(data).to_csv('output.csv', index=False)

这会将两列以所需格式保存到 output.csv。

编辑:上面的答案假定 value1 和 value2 对你没有任何用处，因此可以用列表理解。如果您实际上正在使用这些值进行一些其他计算，它可能会帮助您使用 defaultdict 在使用时更新值字典，然后在最后输出，就像这样:

import pandas as pd
from collections import defaultdict

output_values = defaultdict(list)

for i in range(0, 100):
    value1 = i * 0.01
    value2 = i * 2.75

    output_values['value1'].append(value1)
    output_values['value2'].append(value2)

    # Do other things with these values....

pd.DataFrame.from_dict(output_values).to_csv('output.csv', index=False)