欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

numpy导入数据并挑选出特定的数据组

程序员文章站 2022-04-12 16:14:45
...

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]
相关标签: program python