python - 调整单个子图宽度的简单方法？

Question

aspect='equal' 为长向量绘制几乎一条线，而 ='auto' 将其绘制为矩阵，这也是不需要的。如何强制 vec 显示为 SMALLVEC？我见过相关的 SO，但它们都相当冗长，并且在我的应用程序中添加了很多代码 - 是否有一些简短的内容，例如 ax.set_width()？优选地，ax来自plt.subplots()的axes。

<小时/>

<小时/>

代码:

import matplotlib.pyplot as plt
import numpy as np

KWS = dict(weight='bold', fontsize=14)

def plot_auto(mat, vec):
    fig, axes = plt.subplots(1, 2)
    plt.suptitle("AUTO", **KWS)
    axes[0].imshow(mat, aspect='auto')
    axes[1].imshow(vec, aspect='auto')
    plt.show()

def plot_equal(mat, vec):
    fig, axes = plt.subplots(1, 2)
    plt.suptitle("EQUAL", **KWS)
    axes[0].imshow(mat, aspect='auto')
    axes[1].imshow(vec, aspect='equal')
    plt.show()

def plot_smallvec(smallvec):
    plt.imshow(smallvec)
    plt.title("SMALLVEC", **KWS)

np.random.seed(0)
mat = np.random.randn(500, 500)
vec = np.random.randn(500, 1)
smallvec = np.random.randn(25, 1)

plot_auto(mat, vec)
plot_equal(mat, vec)
plot_smallvec(smallvec)

Answer 1

归功于@ImportanceOfBeingErnest建议使用 aspect 的第三个选项参数(num)。我认为如果您想匹配不同形状数组生成的方面，包含通用公式可能会很有用。公式为

(reference.shape[0] / reference.shape[1]) / (target.shape[0] / target.shape[1])

其中 reference 是您希望匹配其绘制方面的数组，target 是要使用匹配方面绘制的数组。将此包含在您的示例代码中，

# ...
def plot_equal(mat, vec, smallvec):
    fig, axes = plt.subplots(1, 2)
    plt.suptitle("EQUAL", **KWS)
    axes[0].imshow(mat, aspect='auto')
    axes[1].imshow(vec, aspect=(smallvec.shape[0] / smallvec.shape[1]) / 
                   (vec.shape[0] / vec.shape[1]))
    plt.show()

# ...
plot_equal(mat, vec, smallvec)

给予

<小时/>

注意   -   此公式可能会因图形的其他变化而崩溃，但在示例代码的范围内它可以正常工作。