2

我正在构建一个包装器以在 Matplotlib 中生成绘图，并且我希望能够有选择地指定构建绘图的轴。

例如，我有：

def plotContourf(thing, *argv, **kwargs):
    return plt.tricontourf(thing[0], thing[1], thing[2], *argv, **kwargs)

def plotScatter(thing, *argv, **kwargs )
    return plt.scatter(thing[0], thing[1], *argv, **kwargs)

fig, ((ax0,ax1),(ax2,ax3)) = plt.subplots(2,2)

plotContourf(some_thing, axes=ax0)
plotScatter(some_thing, axes=ax2)

哪个运行，但所有内容都绘制在最后一个轴 (ax3) 上，而不是通过轴 kwargument 指定的轴上。（这里没有错误，它只是出现在错误的轴上）

出于好奇，我想这样做的原因是用户可以设置一个轴，或者对于懒惰的人，他们可以在没有指定轴的情况下调用 plotContourf() 并且仍然可以获得他们可以 plt.show( )

另一方面，我尝试了

def plotContourf(thing, axes=None, *argv, **kwargs):
    if axes is None:
        fig, axes = plt.subplots()

    return axes.tricontourf(thing[0], thing[1], thing[2], *argv, **kwargs)

但后来我得到：

TypeError：plotContourf() 为关键字参数“axes”获取了多个值

我知道这个错误是由于“axes”已经是一个关键字参数。我知道我可以使用不同的关键字，但是 axes kwarg 有什么用？

谢谢！

编辑： 完整回溯（对于上述第二个选项）：

Traceback (most recent call last):
  File "mods.py", line 51, in <module>
    adcirc.plotContourf(hsofs_mesh, -1*hsofs_mesh['depth'], axes=ax0)
TypeError: plotContourf() got multiple values for keyword argument 'axes'

而实际的包装器：

def plotContourf(grid, axes=None, *argv, **kwargs): 
    if axes is None:
        fig, axes = plt.subplot()
    return axes.tricontourf(grid['lon'], grid['lat'], grid['Elements']-1, *argv, **kwargs)
4

2 回答 2

1

问题是您使用-1*hsofs_mesh['depth']代替调用该函数，axes然后axes在最后再次添加关键字参数。

In [10]: def fun(a, x=1, *args, **kwargs):
    ...:     print(x)
    ...:     

In [11]: fun(1, 3, x=4)
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-11-31b5e42fb4be> in <module>()
----> 1 fun(1, 3, x=4)

TypeError: fun() got multiple values for keyword argument 'x'

在这个例子中看到它正在阅读3，x然后我正在传递x=4。导致您遇到的错误。

一个解决方案是向您的函数添加另一个参数，如下所示：

def plotContourf(thing, other_thing=None, axes=None, *argv, **kwargs):
于 2017-08-15T18:28:55.183 回答
1

您可以很好地保留axes关键字参数，这样可以省去考虑其他参数顺序的麻烦。

import matplotlib.pyplot as plt
import numpy as np

def plotContourf(thing, *argv, **kwargs):
    axes = kwargs.pop("axes", None)
    if not axes:
        fig, axes = plt.subplots()

    return axes.tricontourf(thing[0], thing[1], thing[2], *argv, **kwargs)

a = [np.random.rand(10) for i in range(3)]
plotContourf(a) 

#or

fig, ax = plt.subplots()
plotContourf(a, axes=ax)

plt.show()
于 2017-08-16T09:30:29.950 回答