在 matplotlib 散点图中排序标签

Question

我有下面的散点图代码，对应的图如下：

x = ['C9-U2', 'C10-U5', 'C10-U5', 'C11-U1', 'C11-U1']
y = ['J',     'C',      'H',      'J',     'H']
plt.scatter(x,y)

在图中，我希望看到两个轴都已排序，即 x 轴应该是 [C9, C10, C11] （就是这样，因为我已经按该顺序输入了数据），并且y 轴应该是 [C, H, J]（不是）。

如何确保两个轴都已排序？

Answer 1

这实际上是一个目前没有好的解决方案的问题。轴单位是根据输入确定的。因此，一种解决方案是手动预先确定分类顺序，先按正确的顺序绘制某些内容，然后再将其删除。

import matplotlib.pyplot as plt

x = ['C9-U2', 'C10-U5', 'C10-U5', 'C11-U1', 'C11-U1']
y = ['J',     'C',      'H',      'J',     'H']

def unitsetter(xunits, yunits, ax=None, sort=True):
    ax = ax or plt.gca()
    if sort:
        xunits = sorted(xunits)
        yunits = sorted(yunits)
    us = plt.plot(xunits, [yunits[0]]*len(xunits),
                  [xunits[0]]*len(yunits), yunits)
    for u in us:
        u.remove()

unitsetter(x,y)
plt.scatter(x,y)

plt.show()

这里，sort 设置为 True，因此您可以在两个轴上获得按字母顺序排序的类别。

如果您有一个希望轴服从的自定义顺序，就像这里的情况（至少对于 x 轴），您需要将该顺序提供给上述函数。

unitsetter(x, sorted(y), sort=False)
plt.scatter(x,y)

Answer 2

在“ImportanceOfBeingErnest”之后，代码可以缩短为

# initial plot to set sorted axis label
us = plt.plot(sorted(x),sorted(y))
[u.remove() for u in us]

# now plot the real thing, sorting not required
plt.scatter(x,y)

Answer 3

我改变了你的散点图的创建方式。

这是我的代码：

import matplotlib.pyplot as plt

# This is your original code.
# x = ['C9-U2', 'C10-U5', 'C10-U5', 'C3-U1', 'C3-U1']
# y = ['J',     'C',      'H',      'J',     'H']
# plt.scatter(x,y)
# plt.show()

ordered_pairs = set([
     ('C9-U2', 'J'),
     ('C10-U5', 'C'),
     ('C10-U5', 'H'),
     ('C3-U1', 'J'),
     ('C3-U1', 'H')
])

x,y = zip(*ordered_pairs)
plt.scatter(x, y)
plt.show()

我将您的数据点转换为有序对的set。这让我们zip 集合，它用于打包和解包每个传递的参数的数组。我们使用* 运算符来逆过程。你可以阅读更多关于ziphere的信息。

当代码运行时，显示的图像如下，我希望这就是你要找的：