用DataFrame的两列的值填充一个列表字典

Question

如果我有一个DataFrame存储2列的值(a &B)从CSV文件，我将如何使用for循环填充字典从DataFrame获取值?

我需要像这样存储A B对的行。A_&_B_sets = {1:[A1,B1]，2:[A2,B2]，…}

A_&_B_sets = {1:[A1,B1],2:[A2,B2],…}
for i in (1,n+1):
A_&_B_sets[i] = i * I

我很迷路。如有任何帮助，不胜感激。

Answer 1

如果您有这样的df:

Column A  Column B
0         1         4
1         2         5
2         3         6

那么你可以这样做:

A_and_B_sets = {}
for i, (_, row) in enumerate(df.iterrows(), 1):
A_and_B_sets[i] = [row["Column A"], row["Column B"]]
print(A_and_B_sets)

打印:

{1: [1, 4], 2: [2, 5], 3: [3, 6]}