主页 / coding / 问题 / 77590201
Accepted
Umar
Asked: 2023-12-02 19:43:11 +0800 CST

Matplotlib 中三角形边框的问题

  • 772

我在 Python 中使用 Matplotlib 绘制三角形边框时遇到问题。我想创建一个特定的模式，但遇到了意外的行为。我需要帮助来识别和解决问题。

这是我的代码

import numpy as np
import matplotlib.pyplot as plt
N = 5
A = np.array([(x, y) for y in range(N, -1, -1) for x in range(N + 1)])
t = np.array([[1, 1], [-1, 1]])
A = np.dot(A, t)

# I have defined a triangle
fig = plt.figure(figsize=(10, 10))
triangle = fig.add_subplot(111)

X = A[:, 0].reshape(N + 1, N + 1)
Y = A[:, 1].reshape(N + 1, N + 1)

for i in range(1, N + 1):
    for j in range(i):
        line_x = np.array([X[i, j + 1], X[i, j], X[i - 1, j]])
        line_y = np.array([Y[i, j + 1], Y[i, j], Y[i - 1, j]])
        triangle.plot(line_y,line_x,  color='black', linewidth=1)

plt.show()

但我得到了这张图片，正如你所看到的，在此输入图像描述

在拐角处，额外的线即将到来，因为我包围了它。我不想要这条额外的线，我尝试使用循环来解决它，尽管一条额外的线将保留

for i in range(6):
    if i == N-1 :
        for j in range(i-1):
            line_x = np.array([X[i , j+1], X[i, j],X[i-1, j]])
            line_y = np.array([Y[i, j+1], Y[i, j], Y[i-1, j]])
            triangle.plot(line_y, line_x, color='black', linewidth=1)
        pass
    else:
        for j in range(i):
            line_x = np.array([X[i , j+1], X[i, j],X[i-1, j]])
            line_y = np.array([Y[i, j+1], Y[i, j], Y[i-1, j]])
            triangle.plot(line_y,line_x, color='black', linewidth=1)
        pass

plt.show() 请解决问题

python

2 个回答

  1. Best Answer

    您要删除的行属于嵌套循环的第一次 (i = 1, j = 0) 和最后一次 (i = N, J = N - 1) 迭代中生成的三角形for：

    import numpy as np
import matplotlib.pyplot as plt

N = 5
A = np.array([(x, y) for y in range(N, -1, -1) for x in range(N + 1)])
t = np.array([[1, 1], [-1, 1]])
A = np.dot(A, t)

# I have defined a triangle
fig = plt.figure(figsize=(10, 10))
triangle = fig.add_subplot(111)

X = A[:, 1].reshape(N + 1, N + 1)
Y = A[:, 0].reshape(N + 1, N + 1)

for i in range(1, N + 1):
    for j in range(i):
        if i == 1: # Lower right triangle
            line_x = np.array([X[i, j + 1], X[i, j]])
            line_y = np.array([Y[i, j + 1], Y[i, j]])
        elif j == N - 1: # Upper right triangle
            line_x = np.array([X[i, j], X[i - 1, j]])
            line_y = np.array([Y[i, j], Y[i - 1, j]])
        else:
            line_x = np.array([X[i, j + 1], X[i, j], X[i - 1, j]])
            line_y = np.array([Y[i, j + 1], Y[i, j], Y[i - 1, j]])
        triangle.plot(line_x, line_y,  color='black', linewidth=1)

plt.show()

    在此输入图像描述

    • 1

  2. 你聪明的方法画了空心三角形，你说空心三角形的最后一列、最右边的列有太多边......

    我有另一种巧妙的方法来绘制“正方形金字塔”，因为这似乎就是您想要的。

    在此输入图像描述

    def pyramid_of_squares(N, ax=None):
    from matplotlib.pyplot import subplots
    from numpy import array
    
    if N<2 : raise ValueError("N must be greater or equal to 2.")
    
    def P(x, y): return array((x, y))
    def draw(p0, p1): ax.plot(*zip(p0, p1), 'k', lw=0.75)
    
    p0 = P(0, 0)
    ptop = P(N-1, N-1)
    pbot = P(N-1, 1-N)
    dp1, dp2 = P(1, -1), P(-1,-1)
    
    if ax is None: fig, ax = subplots(figsize=(4, 7))
    ax.grid()
    ax.set_aspect('equal')

    draw(p0, ptop)
    draw(p0, pbot)    
    for i in range(N-1):
        j = i+1
        draw(p0+dp1*j, ptop+dp1*j+dp2*i)
        draw(p0-dp2*j, pbot-dp2*j-dp1*i)
    return ax

from matplotlib.pyplot import show, subplots
fig, axs = subplots(1, 3, figsize=(9, 4))
for i, ax in enumerate(axs):
    pyramid_of_squares(2+i*3, ax)
    • 1

相关问题