我正在嘗試提取分組的行數據以使用值將其與另一個文件的標簽顏色一起繪制。我的數據框如下所示。df = pd.DataFrame({'x': [1, 4, 5], 'y': [3, 2, 5], 'label': [1.0, 1.0, 2.0]}) x y label0 1 3 1.01 4 2 1.02 5 5 2.0我想獲得一組標簽列表，例如{'1.0': [{'index': 0, 'x': 1, 'y': 3}, {'index': 1, 'x': 4, 'y': 2}], '2.0': [{'index': 2, 'x': 5, 'y': 5}]}這該怎么做？
9 回答
楊__羊羊
TA貢獻1943條經驗 獲得超7個贊
您可以使用itertuples和defulatdict：
itertuples 返回命名元組以迭代數據幀：
for row in df.itertuples():
print(row)
Pandas(Index=0, x=1, y=3, label=1.0)
Pandas(Index=1, x=4, y=2, label=1.0)
Pandas(Index=2, x=5, y=5, label=2.0)
所以利用這一點：
from collections import defaultdict
dictionary = defaultdict(list)
for row in df.itertuples():
dummy['x'] = row.x
dummy['y'] = row.y
dummy['index'] = row.Index
dictionary[row.label].append(dummy)
dict(dictionary)
> {1.0: [{'x': 1, 'y': 3, 'index': 0}, {'x': 4, 'y': 2, 'index': 1}],
2.0: [{'x': 5, 'y': 5, 'index': 2}]}
添加回答
舉報
0/150
提交
取消