pip install matplotlib numpy

import matplotlib.pyplot as plt
import numpy as np

# 創(chuàng)建一個一維數(shù)組
arr_1d = np.array([1, 2, 3, 4, 5])
 
# 創(chuàng)建一個二維數(shù)組
arr_2d = np.array([[1, 2, 3], [4, 5, 6]])
 
# 使用arange函數(shù)生成一個等差數(shù)列
arr_range = np.arange(0, 10, 2)  # 從0到10（不包括10），步長為2

# 數(shù)組的形狀
print(arr_2d.shape)  # 輸出: (2, 3)
 
# 數(shù)組的類型
print(arr_1d.dtype)  # 輸出: int64
 
# 數(shù)組的索引和切片
print(arr_2d[0, 1])  # 輸出: 2
print(arr_2d[0, :])  # 輸出: [1 2 3]
 
# 數(shù)組的形狀變換
arr_reshaped = arr_1d.reshape((5, 1))
print(arr_reshaped)
# 輸出:
# [[1]
#  [2]
#  [3]
#  [4]
#  [5]]

# 數(shù)組間的加法
arr_add = arr_1d + 10
print(arr_add)  # 輸出: [11 12 13 14 15]
 
# 數(shù)組間的乘法（元素級乘法）
arr_mul = arr_1d * 2
print(arr_mul)  # 輸出: [ 2  4  6  8 10]
 
# 數(shù)組的平方根
arr_sqrt = np.sqrt(arr_1d)
print(arr_sqrt)  # 輸出: [1.         1.41421356 1.73205081 2.         2.23606798]

x = np.arange(0, 10, 1)
y = x ** 2
 
plt.plot(x, y)
plt.xlabel('x')
plt.ylabel('y')
plt.title('Simple Line Plot')
plt.show()

categories = ['A', 'B', 'C', 'D']
values = [4, 7, 1, 8]
 
plt.bar(categories, values)
plt.xlabel('Categories')
plt.ylabel('Values')
plt.title('Simple Bar Plot')
plt.show()

x = np.random.rand(50)
y = np.random.rand(50)
 
plt.scatter(x, y)
plt.xlabel('x')
plt.ylabel('y')
plt.title('Simple Scatter Plot')
plt.show()

x = np.linspace(0, 2 * np.pi, 100)  # 生成0到2π之間的100個點
y_sin = np.sin(x)
y_cos = np.cos(x)
 
plt.plot(x, y_sin, label='sin(x)')
plt.plot(x, y_cos, label='cos(x)')
plt.xlabel('x')
plt.ylabel('y')
plt.title('Sine and Cosine Functions')
plt.legend()  # 顯示圖例
plt.show()

data = np.random.randn(1000)  # 生成1000個服從正態(tài)分布的隨機數(shù)
 
plt.hist(data, bins=30, edgecolor='black')
plt.xlabel('Value')
plt.ylabel('Frequency')
plt.title('Histogram of Normal Distribution')
plt.show()

from mpl_toolkits.mplot3d import Axes3D
 
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
 
x = np.random.rand(50)
y = np.random.rand(50)
z = np.random.rand(50)
 
ax.scatter(x, y, z, c='r', marker='o')
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')
ax.set_title('3D Scatter Plot')
plt.show()

x = np.arange(0, 10, 1)
y1 = x ** 2
y2 = np.sqrt(x)
 
plt.subplot(2, 1, 1)  # 2行1列，當(dāng)前是第1個子圖
plt.plot(x, y1)
plt.xlabel('x')
plt.ylabel('y1')
plt.title('y1 = x^2')
 
plt.subplot(2, 1, 2)  # 2行1列，當(dāng)前是第2個子圖
plt.plot(x, y2)
plt.xlabel('x')
plt.ylabel('y2')
plt.title('y2 = sqrt(x)')
 
plt.tight_layout()  # 自動調(diào)整子圖參數(shù), 使之填充整個圖像區(qū)域
plt.show()