如何在Python中将绘图数据保存到输出数据文件中

import matplotlib.pyplot as plt

# 绘制图形
plt.plot([1, 2, 3, 4], [1, 4, 9, 16])

# 保存图形为PNG文件
plt.savefig('output.png')

import pandas as pd

# 创建绘图数据
data = {'x': [1, 2, 3, 4], 'y': [1, 4, 9, 16]}

# 创建DataFrame对象
df = pd.DataFrame(data)

# 保存DataFrame为Excel文件
df.to_excel('output.xlsx', index=False)

from openpyxl import Workbook
from openpyxl.chart import LineChart, Reference

# 创建工作簿和工作表
wb = Workbook()
ws = wb.active

# 添加绘图数据
data = [
    [1, 1],
    [2, 4],
    [3, 9],
    [4, 16]
]
for row in data:
    ws.append(row)

# 创建折线图
chart = LineChart()
values = Reference(ws, min_col=2, min_row=1, max_col=2, max_row=4)
categories = Reference(ws, min_col=1, min_row=2, max_row=5)
chart.add_data(values, titles_from_data=True)
chart.set_categories(categories)

# 将图表添加到工作表
ws.add_chart(chart, "D1")

# 保存工作簿为Excel文件
wb.save('output.xlsx')