Python:Pandas:从数据框制作字典

Question

我正在尝试将数据框转换为字典:

xtest_cat = xtest_cat.T.to_dict().values()

但它给出警告:

Warning: DataFrame columns are not unique, some columns will be omitted python

我检查了数据框的列名(xtest_cat):

len(list(xtest_cat.columns.values)) 
len(set(list(xtest_cat.columns.values)))

它们都是独一无二的。 谁能帮我吗 ？

Answer 1

您可以使用 reset_index用于创建 unique 索引:

xtest_cat = pd.DataFrame({'A':[1,2,3],
                          'B':[4,5,6],
                          'C':[7,8,9]})

xtest_cat.index = [0,1,1]
print (xtest_cat)
   A  B  C
0  1  4  7
1  2  5  8
1  3  6  9

print (xtest_cat.index.is_unique)
False

xtest_cat.reset_index(drop=True, inplace=True)
print (xtest_cat)
   A  B  C
0  1  4  7
1  2  5  8
2  3  6  9

xtest_cat = xtest_cat.T.to_dict().values()
print (xtest_cat)
dict_values([{'B': 4, 'C': 7, 'A': 1}, {'B': 5, 'C': 8, 'A': 2}, {'B': 6, 'C': 9, 'A': 3}])

您也可以省略T并添加参数orient='index':

xtest_cat = xtest_cat.to_dict(orient='index').values()
print (xtest_cat)
dict_values([{'B': 4, 'C': 7, 'A': 1}, {'B': 5, 'C': 8, 'A': 2}, {'B': 6, 'C': 9, 'A': 3}])

orient='record' 更好:

xtest_cat = xtest_cat.to_dict(orient='records')
print (xtest_cat)
[{'B': 4, 'C': 7, 'A': 1}, {'B': 5, 'C': 8, 'A': 2}, {'B': 6, 'C': 9, 'A': 3}]