Python计算等差数列前n项和报错：不支持的操作数问题求助

错误原因 #

你第4行的z[1]-z[0]报错，是因为input()获取的内容是字符串，转成list后每个元素都是字符（比如输入123，z会变成['1','2','3']），字符之间不能直接做减法运算，Python不支持这种操作，所以抛出“不支持的操作数”错误。

修复步骤及代码 #

调整输入逻辑，让用户明确输入等差数列的前两项（用逗号分隔，比如2,5），再将输入内容转为数值类型计算：

# 输入等差数列前两项，用逗号分隔
input_str = input('输入等差数列的前两项（格式：首项,第二项，比如2,5）:')
# 分割字符串并转为整数
a, second_term = map(int, input_str.split(','))
n = int(input("输入n的值:"))
d = second_term - a  # 计算公差d
Sn = n/2 * (2*a + (n-1)*d)  # 等差数列前n项和公式
print(Sn)

额外补充 #

• 如果需要支持带小数的等差数列（比如1.5,3.5），把map(int,...)改成map(float,...)即可。
• 原代码的输入方式（直接输入一串字符）会把每个字符当成独立元素，完全不符合等差数列项的输入逻辑，所以必须改成明确输入前两项的形式。

内容的提问来源于stack exchange，提问作者imhere