python - 当“if”在 Pandas Python CSV 中传递时，从同一行中的另一列获取值

Question

我正在尝试将来自不同文件的代码集成到 csv 中。

我搜索文档，但无法真正理解 Pandas 库以使我的代码正常工作。

CSV 表：

ID     | Quantity
-------+---------
235235 | 5
125772 | 4
647454 | 2
236344 | 8

我有一个完整的 ID 列表。我想将 CSV 中的 ID 与另一个列表中的 ID 匹配，并获取Quantity同一行列的确切值。

例如：

236344 = 8

如果我从 id_list 获取 236344 id，则该函数正在检查它是否与 csv 文件中的匹配项，并给出正确的值Quantity

125772 = 4

（简化）：

id_list = open('')
csv_list = pandas.read_csv('', header=0 )

for id_1 in id_list:
  for id_2 in csv_list:
    if( id_1 == id_2 ):
       quantity_value = (value_from_csv)

我需要将其后的数量值传递给另一个函数。

我尝试使用df.loc[]，但无法正常工作。有任何想法吗？

标签： pythonpython-3.xpandaslistcsv