学习笔记：numpy取数据

import numpy as np

x1 = np.arange(10)

# 索引取数
print(x1[0])
print(x1[1:5])
print(x1[[1, 2, 3]])
"""
0
[1 2 3 4]
[1 2 3]
"""

x2 = np.arange(10).reshape(2, 5)
print(x2[0])
print(x2[0, :2])
"""
[0 1 2 3 4]
[0 1]
"""