在列表中存储具有特定值的 DataFrame 行

Question

我有一个像这样的数据框：

id	country	city	amount	duplicated
1	France	Paris	200	1
2	France	Paris	200	1
3	France	Lyon	50	2
4	France	Lyon	50	2
5	France	Lyon	50	2

我想重复存储每个不同值的列表，例如：

列表 1

[
    {
        "id": 1,
        "country": "France",
        "city": "Paris",
        "amount": 200,
    },
    {
        "id": 2,
        "country": "France",
        "city": "Paris",
        "amount": 200,
    }
  ]

清单 2

[
    {
        "id": 3,
        "country": "France",
        "city": "Lyon",
        "amount": 50,
    },
    {
        "id": 4,
        "country": "France",
        "city": "Lyon",
        "amount": 50,
    },
    {
        "id": 5,
        "country": "France",
        "city": "Lyon",
        "amount": 50,
    }
  ]

我尝试过滤重复项

df[df.duplicated(['country','city','amount', 'duplicated'], keep = False)]

但它只返回相同的 df。

Answer 1

如果我理解正确的话，你可以使用DataFrame.to_dict('records') 来制作你的列表：

list_1 = df[df['duplicated'] == 1].to_dict('records')
list_1 = df[df['duplicated'] == 2].to_dict('records')

或者对于列中任意数量的值，您可以创建一个字典：

result = {}
for value in df['duplicated'].unique():
    result[value] = df[df['duplicated'] == value].to_dict('records')