Python中带有字符串的3D Scatterplot

Question

我试图用Python在x和y上使用字符串类别（即神经网络的激活函数和求解器）并在z轴上使用浮点数（即NN的精度得分）来制作3D散点图。

以下示例引发错误： ValueError：无法将字符串转换为float：'str1'

我按照以下文档进行了3D绘图：https://matplotlib.org/mpl_toolkits/mplot3d/tutorial.html

任何想法，可能是什么问题？ 提前非常感谢！

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
xs=['str1', 'str2']
print(type(xs))
ys=['str3', 'str4']
print(type(ys))
zs=[1,2]
ax.scatter(xs, ys, zs)

Answer 1

您试图将类别值（字符串）作为x和y参数传递。这将适用于1d散点图，但在3d中，您需要定义跨度/笛卡尔坐标。您主要希望拥有的字符串是x和y轴刻度标签。要获得所需的图，您可以做的是首先绘制数字值，然后根据您的字符串值重新分配刻度线标签。

placeholder for x

您还可以使用

设置刻度标签

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

xs=['str1', 'str2']
ys=['str3', 'str4']
zs=[1,2]

ax.scatter(range(len(xs)), range(len(xs)), zs)
ax.set(xticks=range(len(xs)), xticklabels=xs,
       yticks=range(len(xs)), yticklabels=xs) 

不过，使用plt.xticks(range(len(xs)), xs) plt.yticks(range(len(ys)), ys) 的第一个选项允许您一行执行相同操作。