python循环遍历各列_如何在Python中循环遍历列？

环行列How do I loop through each column in the 2D array?# zip(*matrix) generates a transposed version of your matrix

for column in zip(*matrix):

do_something(column)

对你提出的问题/例子的回答I would want to do a method that checks if there's at least one column

in the 2D array that the column has the same values

一般方法：def check(matrix):

for column in zip(*matrix):

if column[1:] == column[:-1]:

return True

return False

一行：arr = [[2,0,3],[4,2,3],[1,0,3]]

any([x[1:] == x[:-1] for x in zip(*arr)])

说明：arr = [[2,0,3],[4,2,3],[1,0,3]]

# transpose the matrix

transposed = zip(*arr) # transposed = [(2, 4, 1), (0, 2, 0), (3, 3, 3)]

# x[1:] == x[:-1] is a trick.

# It checks if the subarrays {one of them by removing the first element (x[1:])

# and the other one by removing the last element (x[:-1])} are equals.

# They will be identical if all the elements are equal.

equals = [x[1:] == x[:-1] for x in transposed] # equals = [False, False, True]

# verify if at least one element of 'equals' is True

any(equals) # True

更新01

@BenC写道："You could also skip the [] around the list comprehension so that any

just gets a generator that can be stopped early once/if it returns

false"

所以：arr = [[2,0,3],[4,2,3],[1,0,3]]

any(x[1:] == x[:-1] for x in zip(*arr))

更新02

您还可以使用集合(与@HelloV的答案合并)。

一行：arr = [[2,0,3],[4,2,3],[1,0,3]]

any(len(set(x))==1 for x in zip(*arr))

一般方法：def check(matrix):

for column in zip(*matrix):

if len(set(column)) == 1:

return True

return False

一个集合没有重复的元素，所以如果你将一个列表转换成一个集合set(x)，任何重复的元素都会消失，所以，如果所有元素都相等，那么结果集合的长度等于一个len(set(x))==1。