根据pandas数据框绘制3d散点图，并按组绘制颜色

pip install pandas
pip install matplotlib

import pandas as pd
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# 创建示例数据框
data = {'x': [1, 2, 3, 4, 5],
        'y': [6, 7, 8, 9, 10],
        'group': ['A', 'B', 'A', 'B', 'A']}
df = pd.DataFrame(data)

# 定义颜色映射
colors = {'A': 'red', 'B': 'blue'}

# 创建3D图形对象
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 根据组别绘制散点图
for index, row in df.iterrows():
    ax.scatter(row['x'], row['y'], 0, c=colors[row['group']], marker='o')

# 设置坐标轴标签
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

# 显示图形
plt.show()