借助if else()函数将第一组的数据点绘制成红色,将第二组的数据点绘制成蓝色, 并显
时间: 2023-10-27 12:03:05 浏览: 77
根据函数值,显示不同的颜色
5星 · 资源好评率100%
要借助if else()函数将第一组的数据点绘制成红色,将第二组的数据点绘制成蓝色,我们可以使用Python中的matplotlib库来实现。
首先,我们需要导入matplotlib库,并创建一个新的图表对象和子图对象。然后,我们可以使用if else语句来判断每个数据点属于哪一组,并设置相应的颜色。
假设我们有两组数据,每个组有x和y坐标。
代码示例:
```
import matplotlib.pyplot as plt
# 第一组数据
x1 = [1, 2, 3, 4]
y1 = [2, 4, 1, 3]
# 第二组数据
x2 = [1, 2, 3, 4]
y2 = [3, 1, 4, 2]
# 创建图表和子图对象
fig, ax = plt.subplots()
# 绘制第一组数据点
for i in range(len(x1)):
if i % 2 == 0:
ax.scatter(x1[i], y1[i], color='red') # 将第一组数据点绘制为红色
else:
ax.scatter(x1[i], y1[i], color='blue') # 将第一组数据点绘制为蓝色
# 绘制第二组数据点
for i in range(len(x2)):
if i % 2 == 0:
ax.scatter(x2[i], y2[i], color='red') # 将第二组数据点绘制为红色
else:
ax.scatter(x2[i], y2[i], color='blue') # 将第二组数据点绘制为蓝色
# 显示图表
plt.show()
```
在这段代码中,我们使用一个for循环遍历每个数据点。通过使用i % 2 == 0条件判断,我们可以将偶数索引的数据点绘制成红色,将奇数索引的数据点绘制成蓝色。
最后,我们使用plt.show()函数来显示图表。
阅读全文