Question

我需要读取一个包含一行整数（13 34 14 53 56 76）的输入文件（input.txt），然后计算每个数字的平方和。

这是我的代码：

# define main program function
def main():
    print("\nThis is the last function: sum_of_squares")
    print("Please include the path if the input file is not in the root directory")
    fname = input("Please enter a filename : ")
    sum_of_squares(fname)

def sum_of_squares(fname):
    infile = open(fname, 'r')
    sum2 = 0
    for items in infile.readlines():
        items = int(items)
        sum2 += items**2
    print("The sum of the squares is:", sum2)
    infile.close()

# execute main program function
main()

如果每个数字都在它自己的行上，它可以正常工作。

但是，当所有数字都在以空格分隔的一行时，我无法弄明白该怎么做。在这种情况下，我收到错误：ValueError: invalid literal for int() with base 10: '13 34 14 53 56 76'

Answer 1

您可以使用file.read()获取字符串，然后使用str.split按空格分割。

您需要先将string中的每个数字转换为int，然后使用内置的sum函数计算总和。

另外，您应该使用with语句为您打开和关闭文件：

def sum_of_squares(fname):

    with open(fname, 'r') as myFile: # This closes the file for you when you are done
        contents = myFile.read()

    sumOfSquares = sum(int(i)**2 for i in contents.split())
    print("The sum of the squares is: ", sumOfSquares)

输出：

The sum of the squares is: 13242

Answer 2

您正尝试将字符串与空格一起转换为整数。

您要做的是使用拆分方法（此处为repo.git.log('--reverse')，该方法将返回包含数字的列表字符串，这次没有任何空格。然后，您将遍历此列表，将每个元素转换为 int ，就像您已经尝试过的那样。

我相信你会发现下一步该做什么。：）

这是一个简短的代码示例，使用更多pythonic方法来实现您的目标。

items.split(' ')

Answer 3

您可以尝试使用split()功能在空间上分割您的项目。

来自文档：例如，' 1 2 3 '.split()会返回['1', '2', '3']。

def sum_of_squares(fname):
    infile = open(fname, 'r')
    sum2 = 0
    for items in infile.readlines():
        sum2 = sum(int(i)**2 for i in items.split())
    print("The sum of the squares is:", sum2)
    infile.close()

Answer 4

保持简单，不需要任何复杂的事情。这是一个评论一步一步的解决方案：

def sum_of_squares(filename):

    # create a summing variable
    sum_squares = 0

    # open file
    with open(filename) as file:

        # loop over each line in file
        for line in file.readlines():

            # create a list of strings splitted by whitespace
            numbers = line.split()

            # loop over potential numbers
            for number in numbers:

                # check if string is a number
                if number.isdigit():

                    # add square to accumulated sum
                    sum_squares += int(number) ** 2

    # when we reach here, we're done, and exit the function
    return sum_squares

print("The sum of the squares is:", sum_of_squares("numbers.txt"))

哪个输出：

The sum of the squares is: 13242

如何使用Python 3中的readlines读取由空格分隔的整数输入文件？

4 个答案: