numpy记录

2019-05-28  本文已影响0人  两分与桥
t = numpy.arange(12).reshape(3, 4)

print(t)
# [[0  1  2  3]
#  [4  5  6  7]
# [8 9 10 11]]

# 取行
print(t[2])
# [[8  9 10 11]]

# 取连续的多行
print(t[1:])
# [[4  5  6  7]
#  [8  9 10 11]]

# 取不连续的多行
print(t[[0,2]])
# [[0  1  2  3]
#  [8  9 10 11]]

print(t[1,:])
# [4 5 6 7]

print(t[1:,:])
# [[4  5  6  7]
#  [8  9 10 11]]

# 取列
print(t[:,0])
# [0 4 8]

# 取连续的多列
print(t[:,2:])
# [[2  3]
#  [6  7]
# [10 11]]

# 取不连续的多列
print(t[:,[0,2]])
# [[0  2]
#  [4  6]
# [8 10]]

# 取行和列,取第2行,第3列的值
a = t[1,2]
print(a)       # 6
print(type(a)) # <class 'numpy.int32'>

# 取多行和多列,取第2行到第3行,第2列到第3列的结果
# 取的是行和列交叉点的位置
b = t[1:3, 1:3]
print(b)
# [[5  6]
#  [9 10]]

# 取多个不相邻的点 (行,列)
# 选出来的结果是(0,0) (2,1) (2,3)
c = t[[0, 2, 2], [0, 1, 3]]
print(c)
# [0  9 11]
上一篇 下一篇

猜你喜欢

热点阅读