熊猫查询单输出答案

【问题标题】：pandas query single output熊猫查询单输出
【发布时间】：2016-04-07 15:02:30
【问题描述】：

我有一个 pandas 数据框（称为 smalls），它被多次重新利用以从数据集中创建多个网络图。我正在尝试根据实体类型在其中一个图中设置节点的颜色，并且需要查询原始数据框。但是，当我这样做时，它会产生一个系列，然后我无法对其进行比较。如何修改下面的第一行，只给我数据框中的第一个条目（所有其他条目都相同）？

temp=smalls.Role[smalls.Entity==big_nodes_order[i]]

print(temp)
10    Threat
11    Threat
12    Threat
Name: Role, dtype: object

【问题讨论】：

标签： python python-3.x pandas dataframe

【解决方案1】：

我认为你可以使用iloc 或iat：

temp=smalls.Role[smalls.Entity==big_nodes_order[i]]
print temp
10    Threat
11    Threat
12    Threat
Name: Role, dtype: object

print temp.iloc[0]
Threat

print temp.iat[0]
Threat

print temp.iloc[:1]
10    Threat
Name: Role, dtype: object

【讨论】：

谢谢@jezrael，我不知道为什么我没有想到把答案当作一个系列来处理。我非常专注于试图让 temp 的值成为我忽略了显而易见的一件事。非常感谢。

【解决方案2】：

您也可以使用.head() 方法：

temp=smalls.Role[smalls.Entity==big_nodes_order[i]].head(1)

【讨论】：