python使用matplotlib从txt文本中读取数据并绘制直方图和折线图

python使用matplotlib从txt文本中读取数据并绘制直方图和折线图

import random
import matplotlib.pyplot as plt

# 生成y轴随机数据
y_random = [round(random.uniform(-10, 10), 1) for _ in range(20)]

# 数据写入txt
with open("data.txt", "w", newline="", encoding="utf-8") as datafile:
    content = ' '.join(map(str, y_random))
    datafile.write(content)

# 从txt读出数据
with open("data.txt", "r", newline="", encoding="utf-8") as datafile:
    content = datafile.read()
    y_list = [float(y) for y in content.split()]

x_list = list(range(20))

fig, ax = plt.subplots(1,2)

# 绘制直方图
ax[0].bar(x_list[0:9], y_list[0:9], facecolor='#ff9999', edgecolor='white')
ax[0].set_xticks(x_list[0:9])

for x, y in zip(x_list[0:9], y_list[0:9]):
    if y>=0:
        ax[0].text(x, y+0.01, '%.1f' % y, ha='center', va='bottom')
    else:
        ax[0].text(x, y-0.01, '%.1f' % y, ha='center', va='top')


# 绘制折线图
ax[1].plot(x_list, y_list, color='y')
ax[1].set_xticks(x_list)

for x, y in zip(x_list, y_list):
    if y>=0:
        ax[1].text(x, y+0.01, '%.1f' % y, ha='center', va='bottom')
    else:
        ax[1].text(x, y-0.01, '%.1f' % y, ha='center', va='top')

# 显示图像
plt.show()

Q&A

1、刚开始是将y列表直接写入txtdatafile.write(y_random)，再次读出y_list = datafile.read()的数据y_list是str类型的,使用list(y_list)进行强制类型转换会将y_list每一个字符视为list中的一个元素,包括空格 ，[],直接plot的话会和x的长度不匹配。
2、改为使用如下方式存储和读取：

    #存储
    content = ' '.join(map(str, y_random))
    datafile.write(content)
···
    #读取
    content = datafile.read()
    y_list = content.split()

能够plot出图像但是y轴的数据混乱，并不是升序排序的，如图：

研究发现是list中的元素是str类型的，将list中每一个元素都转换成float就可以了。