plt.contourf实例
时间: 2023-06-12 18:08:19 浏览: 98
这里是一个使用`plt.contourf`函数绘制等高线的简单实例:
```python
import numpy as np
import matplotlib.pyplot as plt
# 创建数据
x = np.linspace(-10, 10, 100)
y = np.linspace(-10, 10, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))
# 绘制等高线
plt.contourf(X, Y, Z, cmap='hot')
# 添加颜色条
plt.colorbar()
# 显示图像
plt.show()
```
这个例子中,我们先使用`np.linspace`函数生成从-10到10之间的100个等间隔的值作为x和y轴的坐标点。然后使用`np.meshgrid`函数生成网格点,最后通过一个函数来计算出每个点的高度值,这里我们使用了`np.sin`函数。最后,我们使用`plt.contourf`函数绘制出等高线图,并使用`cmap`参数指定色图,使用`plt.colorbar`函数添加颜色条。最后使用`plt.show`函数显示图像。
相关问题
给出contourf的实例代码
Contourf实例代码示例:import matplotlib.pyplot as plt
import numpy as npX = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)plt.contourf(X, Y, Z)
plt.colorbar()
plt.show()
逻辑回归python实例
好的,下面是一个简单的逻辑回归 Python 实例:
假设我们有一个数据集,其中包含两个特征 X1 和 X2,和一个二元分类标签 Y(0 或 1)。我们需要用逻辑回归模型预测新数据的分类标签。
首先,我们需要导入必要的库:
```python
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
```
然后,我们可以加载数据集,假设数据集保存在名为 "data.csv" 的文件中:
```python
data = pd.read_csv("data.csv")
```
接下来,我们可以将数据集分为训练集和测试集:
```python
X_train, X_test, Y_train, Y_test = train_test_split(data[['X1', 'X2']], data['Y'], test_size=0.3, random_state=42)
```
然后,我们可以创建一个逻辑回归模型并拟合训练数据:
```python
model = LogisticRegression()
model.fit(X_train, Y_train)
```
接着,我们可以使用测试数据进行预测,并计算模型的准确性:
```python
Y_pred = model.predict(X_test)
accuracy = accuracy_score(Y_test, Y_pred)
print("Accuracy:", accuracy)
```
最后,我们可以可视化模型的决策边界:
```python
x_min, x_max = X_train['X1'].min() - 1, X_train['X1'].max() + 1
y_min, y_max = X_train['X2'].min() - 1, X_train['X2'].max() + 1
xx, yy = np.meshgrid(np.arange(x_min, x_max, 0.1),
np.arange(y_min, y_max, 0.1))
Z = model.predict(np.c_[xx.ravel(), yy.ravel()])
Z = Z.reshape(xx.shape)
plt.contourf(xx, yy, Z, alpha=0.4)
plt.scatter(X_train['X1'], X_train['X2'], c=Y_train, s=20, edgecolor='k')
plt.title("Decision Boundary")
plt.xlabel("X1")
plt.ylabel("X2")
plt.show()
```
这就是一个简单的逻辑回归 Python 实例,希望对你有所帮助!
阅读全文