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

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