根据一维数组的内容输出二维数组的某些行(一维数组的个数与二维数组的行数相等)

By sto at 2019-03-13 • 0人收藏 • 111人看过

(1)指定一维数组key,由字母a,c,d,e,f组成

(2)随机生成6*3的二维数据value

(3)当一维数组中值为c时,输出对应二维数组value的那行数组

代码如下所示:


>>>  import numpy as np
>>> key = np.array(['a', 'c', 'e', 'c', 'd', 'f'])
>>> key
array(['a', 'c', 'e', 'c', 'd', 'f'], dtype='<U1')
>>> value=np.random.randn(6,3)
>>> value
array([[-0.36870458,  0.80699353,  0.76417796],
       [-0.06467638,  0.33348069,  1.03931474],
       [-0.15536949,  1.94386599, -0.0184938 ],
       [ 0.1329341 ,  0.51435116,  1.12422473],
       [-0.21765932,  0.38687533, -2.16958013],
       [ 0.13778136, -1.77146051,  0.41858588]])
>>> value[key=='c',:]
array([[-0.06467638,  0.33348069,  1.03931474],
       [ 0.1329341 ,  0.51435116,  1.12422473]])


登录后方可回帖

信息栏
数据人网是数据人学习、交流和分享的平台,专注于从数据中学习,努力发觉数据之洞见,积极利用数据之价值
Loading...