Python中的数组可以很方便地进行平方运算,下面通过代码示例来说明。
# 定义一个数组 arr = [1, 2, 3, 4, 5] # 使用循环遍历数组并求平方 for i in range(len(arr)): arr[i] = arr[i] ** 2 # 输出平方之后的数组 print(arr)
在上面的代码中,我们首先定义了一个数组arr,然后使用for循环遍历数组中的每个元素,并使用**运算符对其进行平方操作,最后将结果重新赋值给原来的数组。
运行上述代码,输出结果为:
[1, 4, 9, 16, 25]
可以看到,数组中的每个元素都被平方了。
除了使用循环遍历数组进行平方运算,还可以使用Python中的列表推导式来简化代码,示例如下:
# 定义一个数组 arr = [1, 2, 3, 4, 5] # 使用列表推导式求平方 arr = [i ** 2 for i in arr] # 输出平方之后的数组 print(arr)
运行上述代码,输出结果与之前相同。
总之,Python中的数组可以很方便地进行平方运算,不管是使用循环遍历还是列表推导式都可以实现。