一个Matplotlib绘图例子

最新推荐文章于 2024-01-21 10:30:00 发布

最小二橙

最新推荐文章于 2024-01-21 10:30:00 发布

阅读量413

点赞数

分类专栏： python 文章标签： python numpy

本文链接：https://blog.csdn.net/rogeryu1234/article/details/54428813

版权

python 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

#!/usr/bin/env python
import sys
import numpy as np
import matplotlib.pyplot as plt
import re
from math import log, sqrt, pi

def read_file(file):
    temp = []
    file.readline()
    for line in file:
        temp.append([line.split(',')[0],log(float(line.split(',')[6]))])
    data = np.array(temp)
    return data

def main():
    data1 = []
    data2 = []
    file1 = open('./Couette_debug.hist')
    file2 = open('./Couette_origin.hist')
    data1 = read_file(file1)
    data2 = read_file(file2)
    plot1 = plt.figure()
    plt.figure(figsize=(6,4.8))
    plt.rc('text', usetex=True)
    plt.rc('font', family='serif')
    plt.ylabel(r" $\textbf{Residual}[\rho E]$")
    plt.xlabel(r"$\textbf{t}$")

    plt.plot(data1[:,0],data1[:,1], 'r-', label = 'preconditioning',linewidth=1.8)
    plt.plot(data2[:,0],data2[:,1], 'b--', label = 'origin',linewidth=1.8)
    plt.legend(loc='best',frameon=True)
    plt.savefig('couette_hist.png')

if __name__ == '__main__':
    main()