04 ndarray 读取和修改

我们可以从 ndarray 中获取想要的任何子集。通常，在机器学习中，你需要使用切片拆分数据，例如将数据集拆分为训练集、交叉验证集和测试集。

一、访问元素

我们可以通过索引和方括号[] 访问和修改ndarray中的数据。

import numpy as np

arr = np.arange(10)

print("arr={}".format(arr))
print('arr第一个元素={}'.format(arr[0]))
print('arr第二个元素={}'.format(arr[1]))
print('arr倒数第一个元素={}'.format(arr[-1]))
print('arr倒数二个元素={}'.format(arr[-2]))

运行结果：

arr=[0 1 2 3 4 5 6 7 8 9]
arr第一个元素=0
arr第二个元素=1
arr倒数第一个元素=9
arr倒数二个元素=8

从秩为2的ndarray中获取元素

import numpy as np

arr = np.arange(10).reshape(2, 5)

print("arr={}".format(arr))

print('arr第一行的元素={}'.format(arr[0]))
print('arr第一行第二列元素={}'.format(arr[0][1