Início / coding / Perguntas / 77590201
Accepted
Umar
Asked: 2023-12-02 19:43:11 +0800 CST

Problema com bordas de triângulos no Matplotlib

  • 772

Estou enfrentando um problema ao desenhar bordas triangulares usando Matplotlib em Python. Quero criar um padrão específico, mas estou encontrando um comportamento inesperado. Preciso de ajuda para identificar e resolver o problema.

este é o meu código

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()

mas estou recebendo esta imagem, como você pode ver,insira a descrição da imagem aqui

Na esquina, linhas extras estão chegando, conforme eu as contornei. Não quero essa linha extra, tentei resolver usando loop, embora uma linha extra continue permanecendo

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() gentilmente resolva o problema

python

2 respostas

  1. Best Answer

    As linhas que você deseja excluir pertencem aos triângulos gerados na primeira (i = 1, j = 0) e na última (i = N, J = N - 1) iterações dos forloops aninhados:

    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()

    insira a descrição da imagem aqui

    • 1

  2. Seu método inteligente desenha triângulos abertos, e você diz que as últimas colunas mais à direita dos triângulos abertos têm lados demais...

    Tenho outro método inteligente para desenhar uma "pirâmide de quadrados", pois é isso que você parece querer.

    insira a descrição da imagem aqui

    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

relate perguntas