Question

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

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

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

Answer 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

Answer 2

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

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

pandas查询单输出

2 个答案: