Question

我正在使用numpy来读取csv文件。我正在尝试添加条件以匹配来自csv文件的记录。我的示例team_file看起来像这样

43596,Team1,50,team1data,id1 
43597,Team2,51,team2data,id2
43598,Team3,50,team2data,id2

以下是代码

import numpy as np
reader = np.genfromtxt(team_file, delimiter=',', usecols=np.arange(1, 3), dtype=None)

我想在column3为50时获取column2。在这个例子中，Team1和Team3应该是输出。

我想在没有编写for循环的情况下使用np.where。有没有办法用numpy来实现这个目标？我不能用熊猫。通过读取器[0] [1]给我的值为50，但我如何实现文件中的所有记录？
感谢任何帮助

Answer 1

In [90]: txt=b"""43596,Team1,50,team1data,id1 
    ...: 43597,Team2,51,team2data,id2
    ...: 43598,Team3,50,team2data,id2
    ...: """
In [92]: data=np.genfromtxt(txt.splitlines(),delimiter=',',usecols=[1,3],dtype=None)
In [93]: data
Out[93]: 
array([[b'Team1', b'team1data'],
       [b'Team2', b'team2data'],
       [b'Team3', b'team2data']],
      dtype='|S9')

无法测试＆＃39; 50＆＃39 ;;没有这样的价值任何＆＃39; usecols＆＃39;。

如果我加载所有列，我可以测试第三个字段＆＃39; 50＆＃39;，选择适当的记录：

In [94]: data=np.genfromtxt(txt.splitlines(),delimiter=',',dtype=None)
In [95]: data
Out[95]: 
array([(43596, b'Team1', 50, b'team1data', b'id1'),
       (43597, b'Team2', 51, b'team2data', b'id2'),
       (43598, b'Team3', 50, b'team2data', b'id2')],
      dtype=[('f0', '<i4'), ('f1', 'S5'), ('f2', '<i4'), ('f3', 'S9'), ('f4', 'S3')])
In [96]: data['f2']
Out[96]: array([50, 51, 50])
In [97]: idx = np.where(data['f2']==50)
In [98]: idx
Out[98]: (array([0, 2], dtype=int32),)
In [99]: data[idx]
Out[99]: 
array([(43596, b'Team1', 50, b'team1data', b'id1'),
       (43598, b'Team3', 50, b'team2data', b'id2')],
      dtype=[('f0', '<i4'), ('f1', 'S5'), ('f2', '<i4'), ('f3', 'S9'), ('f4', 'S3')])
In [100]: data['f1'][idx]
Out[100]: 
array([b'Team1', b'Team3'],
      dtype='|S5')

校正。使用arange(1,3)代替[1,3]来选择列

In [102]: data=np.genfromtxt(txt.splitlines(),delimiter=',',usecols=np.arange(1,
     ...: 3),dtype=None)
In [103]: data
Out[103]: 
array([(b'Team1', 50), (b'Team2', 51), (b'Team3', 50)],
      dtype=[('f0', 'S5'), ('f1', '<i4')])
In [104]: idx = np.where(data['f1']==50)
In [105]: data[idx]
Out[105]: 
array([(b'Team1', 50), (b'Team3', 50)],
      dtype=[('f0', 'S5'), ('f1', '<i4')])

如何使用numpy查找多个记录的位置

1 个答案: