Python:如何用逗号组成数组？

Question

我是Python新手，所以请不要 mock 这个问题......

我的文件中有一些数组，如下所示

100 23 35 44 47 511
100 60 77 68 45 76
100 97 99 89 91 14
100 53 65 

我已阅读该文件并使用以下代码获取每一行，

f = file('new.txt')
lines = f.readlines()
f.close()
results = []
for line in lines:
  print line

但为了将它们视为函数的输入，如下所示，

clf.fit ([[0, 0], [1, 1], [2, 2]], [0, 1, 2])

我认为我需要格式化数组，使每个数组都位于方括号 ([ ]) 中，并在它们之间添加逗号。我需要的最终格式是这样的

clf.fit ([[100,23,35,44,47,511], [100,60,77,68,45,76], [100,97,99,89,91,14]], [100,53,65])

如何实现？

Answer 1

只需将每一行拆分为一个列表并创建一个列表列表(本质上是一个数组)。

final_array = []

with open('new.txt') as f:
    for line in f:
        temp_list = [int(x) for x in line.strip().split()]
        if len(temp_list) > 0:  # don't append an empty list (blank line)
            final_array.append(temp_list)

print final_array

您可能想要进行一些额外的健全性检查，但这已经完成了基本想法。