基于Python面向对象编程实现文件路径输入与图表生成的技术问询

我来帮你搞定这个需求！你想要的是一个可以接收用户输入文件路径、然后生成并展示开盘价图表的Python对象对吧？咱们一步步来调整你的代码，让它更规范、更健壮。

先梳理下原代码的几个小问题： #

• 把import语句放在while True循环里会导致每次循环都重新导入模块，完全没必要，应该放在文件最顶部。
• 目前的代码是函数式的，没有用到面向对象的结构，咱们把它改成类的形式更符合你的需求。
• np.loadtxt的delimiter参数你没写完，得根据你的数据文件分隔符来设置（比如逗号、制表符等）。
• 缺少对文件不存在、数据格式错误等异常的处理，用户输入错路径时程序会直接崩溃。

完整的面向对象实现代码 #

from matplotlib import pyplot as plt
from matplotlib import style
import numpy as np

class PriceGraphGenerator:
    def __init__(self):
        # 初始化图表样式
        style.use('ggplot')
        self.file_path = None
        self.x_data = None
        self.y_data = None

    def get_file_path(self):
        """获取用户输入的文件路径，处理文件不存在的情况"""
        while True:
            try:
                self.file_path = input("请输入数据文件路径：")
                # 先尝试打开文件确认存在
                with open(self.file_path, 'r') as f:
                    break
            except FileNotFoundError:
                print(f"错误：找不到文件 {self.file_path}，请重新输入！")
            except Exception as e:
                print(f"发生未知错误：{e}，请重新输入！")

    def load_data(self, delimiter=','):
        """从文件加载数据，处理数据格式错误"""
        try:
            # unpack=True表示把每一列拆分成单独的数组，这里假设第一列是x轴数据，第二列是开盘价y轴
            self.x_data, self.y_data = np.loadtxt(
                self.file_path, 
                unpack=True, 
                delimiter=delimiter,
                # 如果文件有表头，可以加skiprows=1跳过第一行
                # skiprows=1
            )
            print("数据加载成功！")
        except ValueError as e:
            print(f"数据格式错误：{e}，请确保文件是两列用{delimiter}分隔的数值数据！")
            # 加载失败后重新获取路径
            self.get_file_path()
            self.load_data(delimiter)
        except Exception as e:
            print(f"加载数据时出错：{e}")

    def generate_graph(self):
        """生成并显示图表"""
        if self.x_data is None or self.y_data is None:
            print("请先加载数据！")
            return
        
        plt.figure(figsize=(10, 6))
        plt.plot(self.x_data, self.y_data, label='开盘价', color='blue', linewidth=2)
        plt.title('开盘价走势图表')
        plt.xlabel('时间/序号')
        plt.ylabel('开盘价')
        plt.legend()
        plt.grid(True)
        plt.show()

if __name__ == "__main__":
    # 实例化对象
    graph_generator = PriceGraphGenerator()
    # 获取文件路径
    graph_generator.get_file_path()
    # 加载数据（如果你的文件分隔符不是逗号，比如制表符\t，就传入delimiter='\t'）
    graph_generator.load_data(delimiter=',')
    # 生成图表
    graph_generator.generate_graph()

代码说明： #

• 类结构：PriceGraphGenerator类封装了所有逻辑，包括初始化样式、获取路径、加载数据、生成图表，完全符合面向对象的设计思想。
• 异常处理：针对文件不存在、数据格式错误等常见情况做了处理，程序不会轻易崩溃，还会友好提示用户重新输入。
• 灵活性：load_data方法允许你指定分隔符，适配不同格式的数据文件（比如CSV用逗号，TSV用制表符）。
• 易用性：运行文件后会自动提示输入路径，输入正确后直接生成图表，完全匹配你的需求。

使用注意事项： #

• 确保你的数据文件是两列数值型数据，第一列作为x轴（比如日期转成的序号、时间戳等），第二列是开盘价。
• 如果数据文件有表头，记得在load_data里打开skiprows=1的注释，跳过表头行。
• 确保已经安装了所需的依赖包，没装的话可以运行：pip install matplotlib numpy

内容的提问来源于stack exchange，提问作者Oscar Bomboscar