python中numpy库用法

numpy是Python中用于科学计算的一个重要库，它提供了高效的数组操作和数学函数。下面是numpy库的一些常用用法：

1. 创建数组

可以使用numpy中的array函数创建数组，例如：

import numpy as np

# 创建一维数组
a = np.array([1, 2, 3])
print(a)  # [1 2 3]

# 创建二维数组
b = np.array([[1, 2], [3, 4]])
print(b)
# [[1 2]
#  [3 4]]

2. 数组的属性

可以使用shape、dtype、size等属性获取数组的形状、数据类型和元素个数，例如：

import numpy as np

a = np.array([1, 2, 3])
print(a.shape)  # (3,)
print(a.dtype)  # int64
print(a.size)   # 3

b = np.array([[1, 2], [3, 4]])
print(b.shape)  # (2, 2)
print(b.dtype)  # int64
print(b.size)   # 4

3. 数组的索引和切片

可以使用索引和切片操作获取数组中的元素或子数组，例如：

import numpy as np

a = np.array([1, 2, 3])
print(a[0])     # 1
print(a[1:])    # [2 3]

b = np.array([[1, 2], [3, 4]])
print(b[0, 0])  # 1
print(b[:, 0])  # [1 3]

4. 数组的运算

可以使用numpy中的函数进行数组的运算，例如：

import numpy as np

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

# 数组加法
c = a + b
print(c)  # [5 7 9]

# 数组乘法
d = a * b
print(d)  # [ 4 10 18]

# 数组平方
e = np.square(a)
print(e)  # [1 4 9]