Matplotlib，python，如何找到一條線切割x軸的點

我一直在撰寫一些代碼來求解二次方程并將其繪制在圖形上，我想在線切 x 軸的點處得到一個點，以及顯示在它們下方的點的坐標。 這就是代碼現在所做的。 這就是我想要它做的。

代碼如下：

import matplotlib

matplotlib.use('TkAgg')
import matplotlib.pyplot as plt
import numpy as np
import math


def int_input(prompt):
    while True:
        try:
            variable_name = int(input(prompt))
            return variable_name
        except ValueError:
            print("Please enter a whole number (in digits)")


def float_input(prompt):
    while True:
        try:
            variable_name = float(input(prompt))
            return variable_name
        except ValueError:
            print("Please enter a numeric value (in digits)")


def yes_input(prompt):
    while True:
        variable_name = input(prompt).lower()
        if variable_name in ["y", "yes"]:
            return "yes"
        elif variable_name in ["n", "no"]:
            return "no"
        else:
            print("""Please enter either "y" or "n". """)


print("Quadratic Solver")

while True:

    print("Input below the values of a, b and c from an equation that is of the form : ax2   bx   c = 0")
    a = float_input('a: ')

    b = float_input('b: ')

    c = float_input('c: ')

    # calculate the discriminant
    d = (b ** 2) - (4 * a * c)

    # find two solutions
    try:
        solution1 = (-b - math.sqrt(d)) / (2 * a)
    except ValueError:
        solution1 = "none"

    try:
        solution2 = (-b   math.sqrt(d)) / (2 * a)
    except ValueError:
        solution2 = "none"

    if solution1 == solution2:
        if solution1 == "none":
            print("There are no solutions.")
        else:
            print("There is one solution: x = {0}".format(solution1))

    elif solution1 != solution2:
        print('There are two solutions, x can be either {0} or {1}'.format(solution1, solution2))

    graph = yes_input("Do you want to plot that on a graph? (y/n):  ")
    if graph == "yes":
        x = np.linspace(-6, 6, 50)

        fig = plt.figure(figsize=(7, 4))

        y4 = a * x ** 2   b * x   c
        plt.plot(x, y4, 'g:', label='Degree 2')

        # Add features to our figure
        plt.legend()
        plt.grid(True, linestyle=':')
        plt.xlim([-6, 6])
        plt.ylim([-4, 4])

        plt.title(f'A graph of {a}x2   {b}x   {c} = 0')
        plt.xlabel('x-axis')
        plt.ylabel('y-axis')
        plt.get_current_fig_manager().window.wm_geometry(" 0 0")
        fig.canvas.manager.window.attributes('-topmost', 1)
        # Show plot
        plt.show()
    new_question = yes_input('New question? (y/n): ')
    if new_question == "no":
        break

我在 Windows 作業系統上，正在使用 IntelliJ IDEA 來編輯我的 python 代碼。

uj5u.com熱心網友回復：

要將點添加到二次方程的圖形中，您可以使用帶有圓圈標記的 plt.plot 函式，并使用 plt.text 列印點的坐標。

根據您的代碼查看我的解決方案。它不是最干凈的，但很容易理解（第 86-96 行）。

import matplotlib

matplotlib.use('TkAgg')
import matplotlib.pyplot as plt
import numpy as np
import math

def int_input(prompt):
    while True:
        try:
            variable_name = int(input(prompt))
            return variable_name
        except ValueError:
            print("Please enter a whole number (in digits)")


def float_input(prompt):
    while True:
        try:
            variable_name = float(input(prompt))
            return variable_name
        except ValueError:
            print("Please enter a numeric value (in digits)")


def yes_input(prompt):
    while True:
        variable_name = input(prompt).lower()
        if variable_name in ["y", "yes"]:
            return "yes"
        elif variable_name in ["n", "no"]:
            return "no"
        else:
            print("""Please enter either "y" or "n". """)


print("Quadratic Solver")

while True:

    print("Input below the values of a, b and c from an equation that is of the form : ax2   bx   c = 0")
    a = float_input('a: ')

    b = float_input('b: ')

    c = float_input('c: ')

    # calculate the discriminant
    d = (b ** 2) - (4 * a * c)

    # find two solutions
    try:
        solution1 = (-b - math.sqrt(d)) / (2 * a)
    except ValueError:
        solution1 = "none"

    try:
        solution2 = (-b   math.sqrt(d)) / (2 * a)
    except ValueError:
        solution2 = "none"

    if solution1 == solution2:
        if solution1 == "none":
            print("There are no solutions.")
        else:
            print("There is one solution: x = {0}".format(solution1))

    elif solution1 != solution2:
        print('There are two solutions, x can be either {0} or {1}'.format(solution1, solution2))

    graph = yes_input("Do you want to plot that on a graph? (y/n):  ")
    if graph == "yes":
        x = np.linspace(-6, 6, 50)

        fig = plt.figure(figsize=(7, 4))

        y4 = a * x ** 2   b * x   c
        plt.plot(x, y4, 'g:', label='Degree 2')

        # Add features to our figure
        plt.legend()
        plt.grid(True, linestyle=':')
        plt.xlim([-6, 6])
        plt.ylim([-4, 4])

        # Add dots
        offset = 0.3

        if solution1 == solution2:
            if solution1 != "none":
                plt.plot(solution1, 0 , marker="o", markersize=10, color="black", linestyle="None")
                plt.text(solution1, 0-offset, f'({solution1}, {0})', ha='center', va='center')
        elif solution1 != solution2:
            plt.plot([solution1, solution2], [0, 0], marker="o", markersize=10, color="black", linestyle="None")
            plt.text(solution1, 0-offset, f'({solution1}, {0})', ha='center', va='center')
            plt.text(solution2, 0-offset, f'({solution2}, {0})', ha='center', va='center')


        plt.title(f'A graph of {a}x2   {b}x   {c} = 0')
        plt.xlabel('x-axis')
        plt.ylabel('y-axis')
        plt.get_current_fig_manager().window.wm_geometry(" 0 0")
        fig.canvas.manager.window.attributes('-topmost', 1)
        # Show plot
        plt.show()
    new_question = yes_input('New question? (y/n): ')
    if new_question == "no":
        break

標籤：Python matplotlib

上一篇：如何在kdeplot/displot中為每個色調組設定不同的線型

下一篇：如何更改分組條的默認分組顏色