python - 阅读文本文件

Question

我有这段代码可以将随机数写入文本文件:

import random
members = 5
participants=[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]
random.shuffle(participants)
with open("myfile1.txt",'w') as tf:
    for i in range(len(participants) // members + 1):
         group = participants[i*members:i*members + members]
         for participant in group:
            tf.write(str(participant))
            tf.write("\n")

我试图用它来计算它给我的随机数

import numpy
data = numpy.loadtxt("myfile1.txt")
def MA1001():
    return(data[0:,0].sum())

但是，它给了我“数组索引过多”的错误？有什么办法可以解决这个问题还是有更好的方法？谢谢!

Answer 1

虽然不能 100% 确定您想要实现什么:您在文件中每行写入一个数字，loadtext 将返回一个数组(形状 (20,))。因此， data.sum() 就足够了。没有第二个维度。