如何从文本文件的一列获取值以获取另一列的值

Question

我正在从文本文件读取数据。 文本文件基本上包含这样的值

1 2 5
1 3 5
1 5 8
2 2 10
2 3 5
2 5 4

我的密码

data = np.loadtxt('test.txt')
player = data.T[0]
position = data.T[1]
score = data.T[2]

基本上，如果我想找到玩家1的所有得分，我会score[player==1]这将给我5,5,10。 但是我想找到两个玩家共有的所有得分值，所以如果我执行此score[player==1 or player==2]我会收到一条错误消息：

ValueError：具有多个元素的数组的真值不明确。 使用a.any（）或a.all（）

请指教如何实现这一目标？

Answer 1

这将为您提供玩家1和2结合的所有分数：

>>> score[np.logical_or(player==1, player==2)]
array([  5.,   5.,   8.,  10.,   5.,   4.]) 

如果您正在寻找两个球员之间的分数交集，请使用：

>>> np.intersect1d(score[player==1], score[player==2])
5.0