numpy导入数据并挑选出特定的数据组
导入numpy模块,并把模块重命名为np
import numpy as np
从’文件名.txt’导入数据,跳过第一行,数字间以tab分割:
x = np.genfromtxt('文件名.txt',skip_header=1,delimiter=' ')
逗号前一个代表行,后一个代表列,可以用冒号表示范围,1:2表示从第二行(列)到第三行(列)。1:9:2代表从第二行到第十行,步长为2,即2、4、6、8行。省略号代表所有的行(列)
a=x[...,0] #a是所有行的第1列
下面的没赋值。写成a1=balabala后,即可使用a1。
第三列等于1的时候的状态,结构为[True,False,False…]结构。
x[...,2]==1
第三列等于1时,对应的行:
x[x[...,2]==1] #注意和上一个代码比较
第三列等于1时对应的行的第一列 (就硬绕)
x[x[...,2]==1][...,0]