python中读取文本数据并采用matplotlib画图

1. 画图

 import numpy as np
 import matplotlib.pyplot as plt
 
 data_file = open("depth.txt")
 line = data_file.readline()
 data_list = []
 while line:
       num = list(map(float, line.split()))
       data_list.append(num)
       line = data_file.readline()
 data_file.close()
 data_array = np.array(data_list)
 plt.imshow(data_array)
 plt_title_txt = "depth"
 plt.title(plt_title_txt)
 plt.colorbar()
 plt.show()

运行结果：

2. 将数据以折线形式表示

 import numpy as np
 import matplotlib.pyplot as plt
 
 data_file = open("depth.txt")
 line = data_file.readline()
 data_list = []
 while line:
       num = list(map(float, line.split()))
       data_list.append(num)
       line = data_file.readline()
 data_file.close()
 data_array = np.array(data_list)
 data = data_array.flatten() //将二维数组转为一维数组
 plt.plot(data)
 plt_title_txt = "depth"
 plt.title(plt_title_txt)
 plt.show()

运行结果：