从df到变量的存储值

Question

我试图从数据帧中提取一个值并将其放入变量中。稍后我会将该值记录到Excel工作簿中。在

首先，我运行一个SQL查询并将其存储到df中：

df = pd.read_sql(strSQL, conn)

我正在循环查看另一个项目列表，并在df中查找它们。它们通过df中的MMString和我循环遍历的项列表中的MMConcat连接。在

^{pr2}$

我在上面的最后一行代码中得到了以下错误。ValueError:只能将大小为1的数组转换为Python标量

在调试控制台中，当我运行最后一行代码但没有将其存储到变量中时，我得到的是一个字符串值。例如，“皮卡”。在

如何将在df中查找的值存储到变量中？在

Answer 1

用loc按行和列索引返回序列，然后通过iat提取第一个值：

Category = df.loc[df['MMString'] == MMConcat, 'CategoryName'].iat[0]

或者，从NumPy数组表示中获取第一个值：

^{pr2}$

这些文档没有帮助，但是^{}只是调用^{}，并且只对具有一个值的序列有效：

pd.Series([1]).item()     # 1
pd.Series([1, 2]).item()  # ValueError: can only convert an array of size 1