numpy 模块
numpy中有两个函数可以用来读取文件,主要是txt文件,
第一个是loadtxt, 其一般用法为
numpy.loadtxt(fname, dtype=, comments='#', delimiter=None, converters=None, skiprows=0, usecols=None, unpack=False, ndmin=0)
dtype=np.float32,dtype=int,Python默认的数字的数据类型为双精度浮点数
这里的skiprows是指跳过前1行, 如果设置skiprows=2, 就会跳过前两行
这里的comment的是指, 如果行的开头为#就会跳过该行
这里的usecols是指只使用0,2两列, unpack是指会把每一列当成一个向量输出, 而不是合并在一起。
test.txt
A B C
1, 2, 3
# AA AAA
4, 5, 6
7, 8, 9
import numpy as np
(a, b) = np.loadtxt('test.txt', dtype=int, skiprows=1, comments='#', delimiter=',', usecols=(0, 2), unpack=True)
print(a, b)
输出
[1 4 7] [3 6 9]
a = np.array([22,87,5,43,56,73,55,54,11,20,51,5,79,31,27])
x = np.arange(0, 2 * np.pi, 0.1)
x = np.linspace(0, 2 * np.pi, 100)
y1, y2 = np.sin(x), np.cos(x)
x = np.random.uniform(-1, 1, 4)
x4 = np.arange(2, 10, 2)
y4 = np.random.randn(10)