创建具有不同颜色线的python图

Question

我正在尝试制作一个简单的绘图，它读取存储在文件中的 x、y 值，并为每行使用不同的颜色绘制它们。以下是我的尝试。

## to run ctrl+shift+b

#print("hello world")

import csv
import itertools
import pylab
import numpy as np
import matplotlib.pyplot as plt


f =  open('2016-09-09-22_25_A_initial-Hysteresis.txt', 'r')
f2 =  open('2016-09-09-22_25_F_initial-Hysteresis.txt', 'r')
f3 =  open('2016-09-09-22_25_K_initial-Hysteresis.txt', 'r')
f4 =  open('2016-09-09-22_25_P_initial-Hysteresis.txt', 'r')
f5 =  open('2016-09-09-22_25_U_initial-Hysteresis.txt', 'r')
f6 =  open('2016-09-09-22_26_A_initial-Hysteresis.txt', 'r')
f7 =  open('2016-09-09-22_26_F_initial-Hysteresis.txt', 'r')
f8 =  open('2016-09-09-22_26_K_initial-Hysteresis.txt', 'r')
f9 =  open('2016-09-09-22_26_P_initial-Hysteresis.txt', 'r')
f10 =  open('2016-09-09-22_26_U_initial-Hysteresis.txt', 'r')

x = f.readlines()
x2 = f2.readlines()
x3 = f3.readlines()
x4 = f4.readlines()
x5 = f5.readlines()
x6 = f6.readlines()
x7 = f7.readlines()
x8 = f8.readlines()
x9 = f9.readlines()
x10 = f10.readlines()

vars = [x, x2, x3, x4, x5, x6, x7, x8, x9, x10]
colors = ['b', 'r', 'g', 'c']
cc = itertools.cycle(colors)


datatable = []
i = 0
for content in vars:


    for line_num, line_content in enumerate(content):
        data = line_content.split()
        row = [[], [], [], []]


        for index, num in enumerate(data, start = 0):

            isappendable = False


            try:
                data_attempt = float(num)
                index = int(index)
                row[index] = data_attempt
                if(line_num > 15):
                    isappendable = True

            except ValueError:
                data_attempt = 0

        if(isappendable):
            datatable.append(row)

    i = i + 1

    c = next(cc)
    index = []
    input = []
    output = []
    for n,x in enumerate(datatable):

        input.append(x[0])
        output.append(x[1])

    num_str = str(i)
    name = "DUT" + num_str
    plt.plot(input, output,  label = name)

plt.legend(loc='upper left')

plt.show()

当我执行代码时，我得到以下图像。

这看起来非常接近我想要的（有 10 条线彼此明显重叠）但我遇到的问题是图例中的颜色似乎与图中的颜色不匹配。图例显示了我希望绘图的颜色，但线条都是相同的（即使我放大它们）。我将如何解决这个问题？谢谢

Answer 1

我查看了您的代码，试图找出问题所在，并确定了一些您应该更改的地方。

1. 您可以使用 for 循环代替手动为每个文件创建变量。

2. 使用with 上下文管理器将在您完成文件后进行清理。

3. vars 和 input 是 Python 中的保留关键字，所以虽然您可以使用它们（并且可能会起作用），但最好使用其他关键字。 p>

4. 您可以使用 enumerate，就像您对其他 for 循环所做的那样，而不是使用 i。

5. 您无需指定start = 0，因为无论如何这是默认值。

6. 您不需要执行int(index)，因为索引已经是int。

7. 不要像 x 和 index 那样混淆和重复变量名称，这会使它们更难跟踪。

8. 我认为您的主要问题是您的 datatable，它应该在循环中初始化。否则，您将像 @Bazingaa 所说的那样重新绘制相同的东西。

9. 您使用 enumerate 来声明 n，但您从未使用过它。

10. 您声明了一个 index 数组，但从未使用它。

11. num_str 变量没有作用。

我在下面谈到了其中一些事情：

import csv
import itertools
import pylab
import numpy as np
import matplotlib.pyplot as plt

colors = ['b', 'r', 'g', 'c']
cc = itertools.cycle(colors)


# List of filenames to read
filenames = ["2016-09-09-22_25_A_initial-Hysteresis.txt", "2016-09-09-22_25_F_initial-Hysteresis.txt", ...]

for i, filename in enumerate(filenames):
    with open ("/path/to/filename/"+filename) as f:
        content = f.readlines()

        datatable = []

        for line_num, line_content in enumerate(content):
            data = line_content.split()
            row = [[], [], [], []]


            for index, num in enumerate(data):

                isappendable = False

                try:
                    data_attempt = float(num)
                    row[index] = data_attempt
                    if line_num > 15:
                        isappendable = True

                except ValueError:
                    data_attempt = 0

            if(isappendable):
                datatable.append(row)

        c = next(cc)

        input = []
        output = []

        for x in datatable:
            input.append(x[0])
            output.append(x[1])

        name = "DUT" + i
        plt.plot(input, output,  label = name)

plt.legend(loc='upper left')

plt.show()

希望这会有所帮助！

Answer 2

看起来完整的代码可以简化为几行。

import numpy as np
import matplotlib.pyplot as plt

# List of filenames to read
filenames = ["2016-09-09-22_25_A_initial-Hysteresis.txt", 
             "2016-09-09-22_25_F_initial-Hysteresis.txt",
             # ...
             "2016-09-09-22_26_U_initial-Hysteresis.txt"]

for i, filename in enumerate(filenames):
    inp, outp = np.loadtxt(filename, usecols=(0,1), unpack=True)
    if len(inp) < 15:
        continue
    plt.plot(inp, outp,  label = "DUT" + i)

plt.legend(loc='upper left')

plt.show()