使用来自 DataFrame 的 2 列的值填充列表字典

Question

如果我有一个 DataFrame 存储来自 CSV 文件的 2 列（A 和 B）的值，我将如何使用 for 循环填充字典以从 DataFrame 中获取值？

我需要像这样存储 AB 对的行.. 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]}