将 Pandas DataFrame 转换为字典

小能豆

将 Pandas DataFrame 转换为字典

python

我有一个包含四列的 DataFrame。我想将此 DataFrame 转换为 Python 字典。我希望第一列的元素为keys，而同一行中其他列的元素为values。

数据框：

    ID   A   B   C
0   p    1   3   2
1   q    4   3   2
2   r    4   0   9

输出应该是这样的：

{'p': [1,3,2], 'q': [4,3,2], 'r': [4,0,9]}

阅读 46

2024-06-27

共1个答案

小能豆

您可以使用set_index和to_dict方法pandas库来实现这一点

import pandas as pd

# Create the DataFrame
data = {
data
'ID': ['p', 'q', 'r'],
        'A': [1, 4, 4],
        'B': [3, 3, 0],
        'C': [2, 2, 9]}

df = pd.DataFrame(data)

# Set the 'ID' column as the index and convert to dictionary
result = df.set_index('ID').T.to_dict('list')

print(result)

这将为您提供所需的输出：

{'p': [1, 3, 2], 'q': [4, 3, 2], 'r': [4, 0, 9]}

2024-06-27