[英]Displaying Radio buttons horizontally in matplotlib
我正在使用matplotlib.widgets
在我的小部件中创建单选按钮,即将出现的按钮垂直堆叠,我希望它们水平堆叠。
MVCE:
import matplotlib.pyplot as plt
from matplotlib.widgets import RadioButtons
plt.subplots_adjust(left=0.2)
rax = plt.axes([0.5,0.05,0.1,0.1])
radio = RadioButtons(rax ,['1','2','3'], active=0, activecolor='blue' )
plt.show()
我想知道有没有办法水平堆叠这些单选按钮。
当前正在尝试向PR#13374中的 RadioButtons
引入orientation
参数。 尚未完成。
正如我在此PR中评论的那样,另一种选择是对按钮使用散点图。 下面显示了我如何想象此实现。 与常规按钮相比,有两个主要增强功能:
这是通过在内部创建图例来实现的,该图例具有所有可用的必需选项。 “ Legend
任何有效参数也可以用于“按钮”。
import matplotlib.pyplot as plt
from matplotlib.widgets import AxesWidget, RadioButtons
class MyRadioButtons(RadioButtons):
def __init__(self, ax, labels, active=0, activecolor='blue', size=49,
orientation="vertical", **kwargs):
"""
Add radio buttons to an `~.axes.Axes`.
Parameters
----------
ax : `~matplotlib.axes.Axes`
The axes to add the buttons to.
labels : list of str
The button labels.
active : int
The index of the initially selected button.
activecolor : color
The color of the selected button.
size : float
Size of the radio buttons
orientation : str
The orientation of the buttons: 'vertical' (default), or 'horizontal'.
Further parameters are passed on to `Legend`.
"""
AxesWidget.__init__(self, ax)
self.activecolor = activecolor
axcolor = ax.get_facecolor()
self.value_selected = None
ax.set_xticks([])
ax.set_yticks([])
ax.set_navigate(False)
circles = []
for i, label in enumerate(labels):
if i == active:
self.value_selected = label
facecolor = activecolor
else:
facecolor = axcolor
p = ax.scatter([],[], s=size, marker="o", edgecolor='black',
facecolor=facecolor)
circles.append(p)
if orientation == "horizontal":
kwargs.update(ncol=len(labels), mode="expand")
kwargs.setdefault("frameon", False)
self.box = ax.legend(circles, labels, loc="center", **kwargs)
self.labels = self.box.texts
self.circles = self.box.legendHandles
for c in self.circles:
c.set_picker(5)
self.cnt = 0
self.observers = {}
self.connect_event('pick_event', self._clicked)
def _clicked(self, event):
if (self.ignore(event) or event.mouseevent.button != 1 or
event.mouseevent.inaxes != self.ax):
return
if event.artist in self.circles:
self.set_active(self.circles.index(event.artist))
用作
plt.subplots_adjust(left=0.2)
rax = plt.axes([0.5,0.05,0.4,0.07])
radio = MyRadioButtons(rax ,['1','2','3'], active=0, activecolor='crimson',
orientation="horizontal")
plt.show()
要么
rax = plt.axes([0.2,0.5,0.25,0.1])
radio = MyRadioButtons(rax ,["AA", "BB", "CC", "DD"], ncol=2)
由于 stackexchange 不希望将此作为编辑,因此这是来自 @ImportanceOfBeingErnest 的上述答案的更新版本,适用于最新的 matplotlib 版本(例如 >= v3.3)
import matplotlib.pyplot as plt
from matplotlib.widgets import AxesWidget, RadioButtons
from matplotlib import cbook
class MyRadioButtons(RadioButtons):
def __init__(self, ax, labels, active=0, activecolor='blue', size=49,
orientation="vertical", **kwargs):
"""
Add radio buttons to an `~.axes.Axes`.
Parameters
----------
ax : `~matplotlib.axes.Axes`
The axes to add the buttons to.
labels : list of str
The button labels.
active : int
The index of the initially selected button.
activecolor : color
The color of the selected button.
size : float
Size of the radio buttons
orientation : str
The orientation of the buttons: 'vertical' (default), or 'horizontal'.
Further parameters are passed on to `Legend`.
"""
AxesWidget.__init__(self, ax)
self.activecolor = activecolor
axcolor = ax.get_facecolor()
self.value_selected = None
ax.set_xticks([])
ax.set_yticks([])
ax.set_navigate(False)
circles = []
for i, label in enumerate(labels):
if i == active:
self.value_selected = label
facecolor = activecolor
else:
facecolor = axcolor
p = ax.scatter([],[], s=size, marker="o", edgecolor='black',
facecolor=facecolor)
circles.append(p)
if orientation == "horizontal":
kwargs.update(ncol=len(labels), mode="expand")
kwargs.setdefault("frameon", False)
self.box = ax.legend(circles, labels, loc="center", **kwargs)
self.labels = self.box.texts
self.circles = self.box.legendHandles
for c in self.circles:
c.set_picker(5)
self._observers = cbook.CallbackRegistry()
self.connect_event('pick_event', self._clicked)
def _clicked(self, event):
if (self.ignore(event) or event.mouseevent.button != 1 or
event.mouseevent.inaxes != self.ax):
return
if event.artist in self.circles:
self.set_active(self.circles.index(event.artist))
用作
plt.subplots_adjust(left=0.2)
rax = plt.axes([0.5,0.05,0.4,0.07])
radio = MyRadioButtons(rax ,['1','2','3'], active=0, activecolor='crimson',
orientation="horizontal")
plt.show()
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.