python提取第一行数据_python - 磊leilei - numpy取出三维数组每一行的第一个数字并返回一维数组 - ColdFunction...

Python 2.7

注意最后有用flatten把二维数组转化成一维数组。

import numpy as np

a = np.arange(150).reshape((10, 5, 3))

print(a)

print('---')

print(a[:,:,0].flatten())

[[[ 0 1 2]

[ 3 4 5]

[ 6 7 8]

[ 9 10 11]

[ 12 13 14]]

[[ 15 16 17]

[ 18 19 20]

[ 21 22 23]

[ 24 25 26]

[ 27 28 29]]

[[ 30 31 32]

[ 33 34 35]

[ 36 37 38]

[ 39 40 41]

[ 42 43 44]]

[[ 45 46 47]

[ 48 49 50]

[ 51 52 53]

[ 54 55 56]

[ 57 58 59]]

[[ 60 61 62]

[ 63 64 65]

[ 66 67 68]

[ 69 70 71]

[ 72 73 74]]

[[ 75 76 77]

[ 78 79 80]

[ 81 82 83]

[ 84 85 86]

[ 87 88 89]]

[[ 90 91 92]

[ 93 94 95]

[ 96 97 98]

[ 99 100 101]

[102 103 104]]

[[105 106 107]

[108 109 110]

[111 112 113]

[114 115