python - 在 Python 中从列表列表到文件

Question

想象一组数据，如 APT=[[10.2,5,6],[9.2,-7,8],[5.75,1,2]] 我想导出为标题为 ['APT' ,'WN','我们'].

我希望文件的形状很好(像

APT |西|我们

10.2 | 5 | 6

9.2 | -7 | 8

5.75 | 1 | 2

我尝试过不同的东西，比如

with open("file.csv",'w') as f:
    wri= csv.writer(f)
    wri.writerow(["ATP","WN","WE"])
    for element in zip(*APT):
        wri.writerow(element)

还有 Pandas 的 to_excel 函数，但没有一个能按我的意愿工作。

知道如何解决吗？

提前致谢

PS:抱歉，我不知道如何在 StackOverflow 中添加更好的表格

PS1:我读过How to export a list with different size from python to excel但我没有得到正确的文件形状

Answer 1

Pandas 包有一个专门为您的任务设计的方法:to_csv()。

这是您的代码:

# import module
import pandas as pd
# convert your data to DataFrame
df= pd.DataFrame(APT)
# name columns
df.columns = ['APT','WN','WE']
# save DataFrame as .csv
df.to_csv('apt.csv')