如何将函数中的点转换为列表以绘制完整的图表？

Question

所以我创建了一个函数 plot_line(p1,p2)，它将两个点作为输入参数并绘制它们之间的线。两个输入参数是 指定 x 和 y 坐标的列表或元组，即 p1 =(x1,y1)。现在我想进一步制作一个函数，例如：complete_graph(points)，它需要一个列表 点并在这些点上绘制完整的图形。我试图修改 plot_line 函数，以便它只调用 plot() 而不是 show()。我希望通过循环点并为每一对调用 plot_line 来绘制完整的图形，最后在循环之后调用 show()，我该怎么做？

我现在有这个代码：

import matplotlib.pyplot as plt
from math import sqrt

"Task a)"
#call function tuple
def cor_tup(x, y): #coordinates for tuple
    return(x,y)

def plotline(p1,p2):
    x=(p1[0],p2[0]) #x values
    y=(p1[1],p2[1])#y values
    plt.plot(x, y, "-o", label="x,y") #plot the points in graph
    
x1=int(input('enter first desired x value:'))
y1=int(input('enter first desired y value:'))
x2=int(input('enter second desired x value:'))
y2=int(input('enter second desired y value:'))

p1=cor_tup(x1,y1)
p2=cor_tup(x2,y2)

plotline(p1,p2)
plt.xlabel('x')
plt.ylabel('y')
plt.legend
plt.axis([-5,5,-5,5])
plt.title('Linear function')
plt.show()

情节在最终结果中应该是这样的，但我不明白：

Answer 1

这将使您成为一个漂亮的小房子，里面有一颗星星（并且应该适用于任何元组列表）

import matplotlib.pyplot as plt

def plotline(p1, p2):
    x=(p1[0],p2[0]) #x values
    y=(p1[1],p2[1])#y values
    plt.plot(x, y, "-b", label="x,y") #plot the points in graph

data = [(0,0), (0,1), (1,0), (1,1), (0.5, 1.5)]

for i, xy in enumerate(data):
    for xy2 in data[i+1:]:
        plotline(xy, xy2)

x = [d[0] for d in data]
y = [d[1] for d in data]
plt.plot(x, y, 'o',color='r')
plt.show()

如果您需要从用户那里获取输入，您可以保留您的输入代码。

请注意，我只在内循环中从 data[i+1:] 迭代，并在外循环中通过 enumerate 递增 i 以避免重复行（每个点只能与其他点连接一次）

输出：

Answer 2

我想我做了一些你想要的东西：

import matplotlib.pyplot as plt
from math import sqrt

"Task a)"
#call function tuple
def cor_tup(x, y): #coordinates for tuple
    return(x,y)

def plotline(p1, p2):
    x=(p1[0],p2[0]) #x values
    y=(p1[1],p2[1])#y values
    plt.plot(x, y, "-o", label="x,y") #plot the points in graph


def Line_of_points(totalLines):

    listOfPoints = []
    
    for i in range(1, totalLines+1):
        currentx = int(input("Enter first x point " + str(i) + " : "))
        currenty = int(input("Enter first y point " + str(i) + " : "))
        currentPoint1 = (currentx, currenty)

        currentx = int(input("Enter second x point " + str(i) + " : "))
        currenty = int(input("Enter second y point " + str(i) + " : "))
        currentPoint2 = (currentx, currenty)

        points = (currentPoint1, currentPoint2)
        
        listOfPoints.append(points)   
    
    return listOfPoints


def Draw_All_Points(pointList):
    counter = 0

    while counter < len(pointList):
        if counter <= len(pointList)-1: #so at last point, doesnt access out of list len
            firstPoint = pointList[counter][0]
            secondPoint = pointList[counter][1]
            print(firstPoint, secondPoint)
            

            plotline(firstPoint, secondPoint)
            
        else:
            break

        counter += 1

    
lineOfPoints = Line_of_points(2)
Draw_All_Points(lineOfPoints)



plt.xlabel('x')
plt.ylabel('y')
plt.legend
plt.axis([-5,5,-5,5])
plt.title('Linear function')
plt.show()