上述代码中,set_edgecolor() takes 2 positional arguments but 3 were given,为什么?
时间: 2023-11-07 21:23:50 浏览: 163
这个问题的原因是 `set_edgecolors()` 方法需要传入两个参数:第一个参数是颜色值或者颜色值列表,第二个参数是索引列表,用于指定需要设置颜色的点的索引。而在这个例子中,我们只传入了一个参数,即颜色值列表。因此,Python 解释器认为我们只传入了一个参数,而期望得到两个参数,因此导致了这个错误。
要解决这个问题,我们需要将颜色值列表和索引列表都传入 `set_edgecolors()` 方法中。可以使用 NumPy 的 `repeat()` 函数来生成一个与点的数量相同的索引列表,然后将其作为第二个参数传入 `set_edgecolors()` 方法中。修改后的代码如下:
```python
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
# 生成随机数据
n = 50
x = np.random.rand(n)
y = np.random.rand(n)
z = np.random.rand(n)
# 创建 3D 图形对象
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# 绘制 3D 散点图
scatter = ax.scatter(x, y, z, s=50, c=z, cmap='cool')
# 设置坐标轴标签
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')
# 添加交互功能
def on_move(event):
if event.inaxes == ax:
x, y = event.xdata, event.ydata
ind = np.argmin(np.sqrt((x - x)**2 + (y - y)**2))
scatter.set_sizes([50]*(n-1) + [300])
scatter.set_edgecolors(np.repeat('black', n), np.arange(n))
scatter.set_facecolors(np.repeat('gray', n))
scatter.set_edgecolors('red', ind)
scatter.set_facecolors('orange', ind)
ax.text2D(0, 0, f'({x:.2f}, {y:.2f}, {z[ind]:.2f})', transform=ax.transAxes)
fig.canvas.mpl_connect('motion_notify_event', on_move)
plt.show()
```
你可以看到,我们使用 `np.arange(n)` 生成了一个长度为 n 的索引数组,并将其作为第二个参数传入 `set_edgecolors()` 方法中,以便为所有的点都设置颜色。而对于需要突出显示的点,我们只需要将其索引作为第二个参数传入 `set_edgecolors()` 方法中即可。
阅读全文