0

我想从一个有两列的文本文件中读取数据。这是对之前提出的问题Generate a matlibplot bar chart from two columns of data的后续。虽然我的问题类似于如何在 Matplotlib 中绘制带有图例的多个两列文本文件中的数据？，我无法让我定义的变量与条形图一起使用。目标是从包含两列的文件中读取数据。

我首先尝试使用genfromtxtwhich 在当前代码中被注释掉，但是收到AttributeError: 'numpy.ndarray' object has no attribute 'split'. 这可能是由于以数组类型格式读取的数据：

>>> import numpy as np
>>> data = np.genfromtxt('input-file', delimiter = ' ')
>>> print(data)
[[ 72.   1.]
 [  9.   2.]
 [ 10.  36.]
 [ 74.   6.]
 [  0.  77.]
 [  5.   6.]
 [  6.  23.]
 [ 72.   1.]
 [  9.   2.]
 [ 10.  36.]
 [ 82.   1.]
 [ 74.   6.]
 [  0.  97.]
 [  5.   6.]
 [  6.  23.]
 [ 72.   1.]
 [  9.   2.]
 [ 10.  36.]
 [ 82.   1.]]

接下来，我尝试直接读取数据但接收AttributeError: 'file' object has no attribute 'split'。

import numpy as np
import matplotlib.pyplot as plt

data = open('input-file', 'r')

#data = np.genfromtxt('input-file', delimiter = ' ')

counts = []
values = []

for line in data.split("\n"):
    x, y = line.split()
    values.append(int(x))
    counts.append(int(y))

#bar_width = 0.25
#opacity = 0.4

plt.bar(values, counts, color='g')
plt.ylabel('Frequency')
plt.xlabel('Length')
plt.title('Title')

plt.show()

示例内容来自input-file.

72 1
9 2
10 36
74 6
0 77
5 6
6 23
72 1
9 2
10 36
82 1
74 6
0 97
5 6
6 23
72 1
9 2
10 36
82 1
4

2 回答 2

1

文件对象没有拆分功能。您可以直接读取行而无需split("\n")

for line in data:
于 2013-09-27T16:46:02.560 回答
1

是什么阻止您直接绘制从 返回的 numpy 数组的列np.genfromtext？

data = np.genfromtxt('tmp.dat', delimiter = ' ')

plt.bar(data[:,0], data[:,1], color='g')
plt.ylabel('Frequency')
plt.xlabel('Length')
plt.title('Title')

plt.show()
于 2013-09-27T16:47:22.500 回答