下面是一些NumPy的例子:
1、创建数组:可以使用NumPy来创建各种类型的数组。
例如,以下代码创建一个包含数字0到9的一维数组:
import numpy as np
arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
print(arr)
输出:
[0 1 2 3 4 5 6 7 8 9]
2、数组运算:NumPy提供了各种数组运算,包括加、减、乘、除等。
例如,以下代码将一个数组中的每个元素都乘以2:
import numpy as np
arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
new_arr = arr * 2
print(new_arr)
输出:
[ 0 2 4 6 8 10 12 14 16 18]
3、多维数组:NumPy还支持多维数组。
例如,以下代码创建一个3x3的二维数组:
import numpy as np
arr = np.array([[0, 1, 2], [3, 4, 5], [6, 7, 8]])
print(arr)
输出:
[[0 1 2] [3 4 5] [6 7 8]]
4、数组索引和切片:可以使用索引和切片来访问和修改数组中的元素。
例如,以下代码将数组中的第一个元素修改为100:
import numpy as np
arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
arr[0] = 100
print(arr)
输出:
[100 1 2 3 4 5 6 7 8 9]
5、数组形状:可以使用NumPy来修改数组的形状。
例如,以下代码将一个一维数组转换为一个3x3的二维数组:
import numpy as np
arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8])
new_arr = arr.reshape((3, 3))
print(new_arr)
输出:
[[0 1 2] [3 4 5] [6 7 8]]
这些只是NumPy的一些例子。NumPy提供了各种函数和工具,可以帮助您进行各种数学和科学计算。