matplotlib画图代码

您好，以下是一个简单的使用matplotlib库进行线性回归可视化的示例代码：

import numpy as np
import matplotlib.pyplot as plt

# 生成数据
x = np.linspace(0, 10, 50)
y = 2 * x + 1

# 添加噪声
y_noise = y + np.random.normal(size=50)

# 绘制散点图
plt.scatter(x, y_noise, label="data")

# 计算线性回归参数
a, b = np.polyfit(x, y_noise, 1)

# 绘制回归直线
y_pred = a * x + b
plt.plot(x, y_pred, color="r", label="fit line")

# 添加图例
plt.legend()

# 显示图像
plt.show()

希望能对您有所帮助！

相关问题

pyqt在canvas上matplotlib画图代码

下面是一个示例代码，使用 PyQt5 和 Matplotlib 在 canvas 上画图：

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QSizePolicy
from PyQt5.QtCore import Qt
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure

class MyCanvas(FigureCanvas):
    def __init__(self, parent=None, width=5, height=4, dpi=100):
        fig = Figure(figsize=(width, height), dpi=dpi)
        self.axes = fig.add_subplot(111)
        FigureCanvas.__init__(self, fig)
        self.setParent(parent)
        FigureCanvas.setSizePolicy(self, QSizePolicy.Expanding, QSizePolicy.Expanding)
        FigureCanvas.updateGeometry(self)
        self.plot()
 
    def plot(self):
        import numpy as np
        x = np.array([1, 2, 3, 4, 5])
        y = x ** 2
        self.axes.plot(x, y)
 
class MyApp(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("PyQt5 & Matplotlib")
        self.setGeometry(300, 300, 600, 400)
 
        self.MyCanvas = MyCanvas(self)
        self.setCentralWidget(self.MyCanvas)
 
app = QApplication(sys.argv)
w = MyApp()
w.show()
sys.exit(app.exec_())

这段代码中，我们首先定义了一个 MyCanvas 类，继承自 FigureCanvas。在 MyCanvas 的 __init__ 方法中，我们创建了一个 Figure 对象，并在其中添加了一个子图，然后调用父类构造函数。在 plot 方法中，我们使用 numpy 生成了一组数据，并绘制了曲线图。

接着定义了一个 MyApp 类继承自 QMainWindow，在其中我们创建了一个 MyCanvas 对象，并将其设为中央窗口部件，最后在运行时加载这个应用

python matplotlib画图颜色代码

以下是常用的颜色代码：

- 红色：`r`或`#FF0000`
- 绿色：`g`或`#00FF00`
- 蓝色：`b`或`#0000FF`
- 黄色：`y`或`#FFFF00`
- 青色：`c`或`#00FFFF`
- 洋红色：`m`或`#FF00FF`
- 黑色：`k`或`#000000`
- 白色：`w`或`#FFFFFF`

此外，还可以使用十六进制颜色代码或RGB颜色代码。例如，`#008000`表示绿色，`rgb(255, 0, 0)`表示红色。