Matplotlib中具有可变长度数据的盒图

2022-05-28 00:00:00 python matplotlib boxplot

问题描述

我在一个文本文件中收集了一些数据,想要创建一个盒图。 但此数据文件包含可变长度的行。

1.2, 2.3, 3.0, 4.5  
1.1, 2.2, 2.9

对于相同的长度,我可以只做

PW  = numpy.loadtxt("./learning.dat")  
matplotlib.boxplot(PW.T);

如何处理可变长度数据行?


解决方案

仅使用数组列表或列表。boxplot将接受任何类型的序列(好的,任何带有__len__的序列。它不适用于发电机等)。

例如:

import matplotlib.pyplot as plt
x = [[1.2, 2.3, 3.0, 4.5],
     [1.1, 2.2, 2.9]]
plt.boxplot(x)
plt.show()

如果您询问如何读取您的数据,有很多方法可以做您想做的事情。举个简单的例子:

import matplotlib.pyplot as plt
import numpy as np

def arrays_from_file(filename):
    """Builds a list of variable length arrays from a comma-delimited text file"""
    output = []
    with open(filename, 'r') as infile:
        for line in infile:
            line = np.array(line.strip().split(','), dtype=np.float)
            output.append(line)
    return output

plt.boxplot(arrays_from_file('test.txt'))
plt.show()

相关文章