numpy.loadtxt用法

numpy.loadtxt(fname, dtype=, comments='#', delimiter=None, converters=None, skiprows=0, usecols=None, unpack=False, ndmin=0)

  • frame要读取的文件、文件名或生成器;
  • dtype数据类型,默认float;
  • comment的是指, 如果行的开头为#就会跳过该行;
  • delimiter分隔符,默认是空格;
  • converters是对数据进行预处理的参数, 我们可以先定义一个函数, 这里的converters是一个字典, 表示第零列使用函数add_one来进行预处理;
  • skiprows是指跳过前1行, 如果设置skiprows=2, 就会跳过前两行,默认读取所有行;
  • usecols是指只使用0,2两列;
  • unpack如果是True会把每一列当成一个向量输出, 而不是合并在一起;

举例:

def add_one(x):
    return int(x)+1#注意到这里使用的字符的数据结构
(a, b) = np.loadtxt('test.txt', dtype=int, skiprows=1, converters={0:add_one}, comments='#', delimiter=',', usecols=(0, 2), unpack=True)
print(a, b)

输入:

%%writefile test.txt # 这是用来写入文件的代码
A B C
1 2 3
4 5 6
7 8 9

输出:

[2 5 8] [3 6 9]

参考文献:

【1】numpy中loadtxt 的用法

原文地址:https://www.cnblogs.com/nxf-rabbit75/p/10467671.html