Question

我正在尝试将字典转换为数据框，然后将其导出为 csv 文件，但由于某种原因，当程序导出数据框时，它会更改列和行。

df = pd.read_csv('test.csv')
for i in range(len(df['name'])):
    names.append(df['name'][i])
    balances.append(df['balance'][i])





def acc_creation(name):
    names.append(name)
    dict = {'name': names, 'balance': balances}
    new_df = pd.DataFrame.from_dict(
        dict, orient='index', columns=['name', 'balance'])
    new_df.to_csv('test.csv', encoding='utf-8', sep=',',
                  header=False, na_rep=0, index=True)

然后它停止工作，这是我得到的错误：

Traceback (most recent call last):
  File "/Users/darkmbs/Desktop/python/test.py", line 9, in <module>
    balances.append(df['balance'][i])
  File "/Library/Frameworks/Python.framework/Versions/3.9/lib/python3.9/site-packages/pandas/core/frame.py", line 3024, in __getitem__
    indexer = self.columns.get_loc(key)
  File "/Library/Frameworks/Python.framework/Versions/3.9/lib/python3.9/site-packages/pandas/core/indexes/base.py", line 3082, in get_loc
    raise KeyError(key) from err
KeyError: 'balance'

我想要的输出：

name,balance
xyzz,0 
abc,0

我得到的输出：

name,xyzz,abcd   
balance,0,0

Answer 1

您面临的问题是 names 和 balances 的长度不同。当您调用 acc_creation(name) 时，您将额外的元素添加到 names (names.append(name))，但不会添加到 balance，因此存在长度不匹配。你可以做的是：

import pandas as pd
import numpy as np

def acc_creation(name):
    names.append(name)
    balances.append(0) # Or any value
    new_df = pd.DataFrame(np.column_stack([names, balances]), 
                                   columns=['name', 'balance'])
    new_df.to_csv('test.csv', encoding='utf-8', sep=',',
                  header=True, na_rep=0, index=True)

Answer 2

我相信你只是删除：

 orient='index',

如果您收到该错误消息，请让我们查看一些原始 csv，以便我们对其进行测试并了解原因。

Python将字典转换为数据框并将其导出到csv文件

2 个答案: