通过 matplotlib.pyplot 在 Python 中绘图（计算面积）

Question

我有一个问题\问题。我需要通过从文件中获得的数字（我做了）绘制图表，然后我需要绘制一条连接起点和终点的线，并计算这两条线之间的面积。我尝试了很多变化，但我不知道我怎么能做到..

我正在尝试通过 matplotlib.pyplot 库来实现

在添加“开始和和之间的连接线”之后我应该得到的“数字”，现在我需要黑线和蓝色之间的计算正方形。 PS黑色的有点直:)

这里有一些代码，我的数据文件... http://pastebin.com/g40bAzPR

#!/path/to/python -tt

# numerical data
# python GraphicalPart.py ../dataFile.txt

import sys
import matplotlib.pyplot as plt
import numpy as np

def startDivide(fileName):
 for i in range(1,2):
    inputFile = open(fileName)
    outputFile = open(fileName + "_" + str(i) + "_out.csv", "w")
    floatList = []
    for line in inputFile.readlines():
        data = line.split(" ")
        string = data[i]
        if string.startswith('-'): #remove '-'
            string = string[1:]
        floatList.append(float(string))

    floatList.sort() #sorting the list of data

    for item in floatList:
        outputFile.write("%s\n" % item)

    outputFile.close() 
    inputFile.close()

    data1=np.genfromtxt(fileName + "_" + str(i) + '_out.csv', skip_header=1)
    plt.plot(data1) 
    plt.savefig(fileName + "_" + str(i) + "_.png")
    plt.clf()

def main():
 if len(sys.argv) != 2:
  print "Not enough arguments. *_data.txt file only!"
 else:
  startDivide(sys.argv[1])

if __name__ == "__main__":
 main()

Answer 1

for i in range(1,2) 是一个只迭代一次的循环。也许您计划增加迭代次数？如果是这样，请记住，加载数据一次 比在 for 循环中多次加载更快。您可以使用 np.genfromtxt 和 usecols 参数来指定所需的列。

要找到曲线下的面积，您可以使用 np.trapz。 要找到两条曲线之间的面积，请从下方曲线下方的面积中减去上方曲线下方的面积。假设对角线始终在数据曲线上方：

import sys
import matplotlib.pyplot as plt
import numpy as np

def startDivide(filename):
    data = np.genfromtxt(filename, dtype=None, usecols=[1])
    data = np.abs(data)
    data.sort()
    np.savetxt("{}_1_out.csv".format(filename), data)
    plt.plot(data)
    plt.plot([0,len(data)-1], [data[0], data[-1]])
    plt.savefig("{}_1_.png".format(filename))
    area = np.trapz([data[0], data[-1]], dx=len(data)-1) - np.trapz(data)
    print(area)

if __name__ == "__main__":
    startDivide(sys.argv[1])