大熊猫数据框用于某些列的用法

时间:2020-10-10 09:04:22

标签: python pandas dataframe dictionary

我有一个像这样的大数据框:

       product   price    serial     category    department   origin    
 0     cookies       4      2345    breakfast          food        V
 1     paper       0.5      4556   stationery          work        V
 2     spoon         2      9843      kitchen     household        M

我想转换为字典,但是我只想要一个输出:

   {serial: 2345}{serial: 4556}{serial: 9843} and {origin: V}{origin: V}{origin: M}

其中键是列名,值是值

现在,我尝试使用df.to_dict('values')并选择dic ['origin']并返回我

   {0: V}{1:V}{2:M}

我也尝试过df.to_dict('records'),但它给了我:

   {product: cookies, price: 4, serial:2345, category: breakfast, department:food, origin:V}

我不知道如何仅选择“原始”或“序列”

1 个答案:

答案 0 :(得分:1)

您可以执行以下操作:

serial_dict = df[['serial']].to_dict('r')
origin_dict = df[['origin']].to_dict('r')