如何将多个数组放入一个带有标题 Python 的 csv 文件

Question

我有这样的列表，我从xlsx 文件加载

import pandas as pd
travel_df = pd.read_excel('./item.xlsx')
data = travel_df.to_dict('records')

data 这样的

data = 
[
    {
        'cat': 'A',
        'subCat': 'a1',
    },
    {
        'cat': 'A',
        'subCat': 'a2',
    },
    {
        'cat': 'B',
        'subCat': 'b1',
    },
    {
        'cat': 'B',
        'subCat': 'b2',
    },
    {
        'cat': 'B',
        'subCat': 'b3',
    },
]

我想把它放到CSV 这样的文件中，最好和最快的方法是什么

A     B
--------
a1    b1
a2    b2
      b3

Answer 1

您可以通过DataFrame()方法、pivot()方法和apply()方法来做到这一点：

newdf=pd.DataFrame(data).pivot(columns='cat',values='subCat').apply(lambda x:sorted(x,key=pd.isna))

最后过滤掉NaN's:

newdf=newdf[~newdf.isna().all(1)]

newdf的输出：

cat   A     B
0     a1    b1
1     a2    b2
2     NaN   b3

现在，如果您想将其保存在 csv 文件中，请使用 to_csv() 方法