How do I get the maximum value of a column in a 2D array using Python?

To find the maximum value of a specific column in a 2D array (list of lists) using Python, you can iterate through the rows and access the element at the desired column index. Here’s how you can do it:

1. Basic Approach using a Loop:

This method involves using a `for` loop to traverse each row and keep track of the maximum value seen so far. This approach is straightforward and easy to understand.

def find_max_in_column(array_2d, column_index):
  if not array_2d or column_index < 0 or column_index >= len(array_2d[0]):
    return None # Return None for invalid input
  max_value = array_2d[0][column_index]
  for row in array_2d:
    if row[column_index] > max_value:
      max_value = row[column_index]
  return max_value

# Example usage
array = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
column_to_check = 1
max_in_column = find_max_in_column(array, column_to_check)
print(f"Maximum value in column {column_to_check}: {max_in_column}") # Output: Maximum value in column 1: 8

Explanation:

- First, the function checks for invalid input: empty array or column index out of bounds, returning `None` in these cases.

- It then initializes `max_value` with the element at the specified column of the first row.

- A loop iterates through each row, and if a value in the specified column is greater than the current `max_value`, the `max_value` is updated.

- Finally, the function returns the `max_value` found.

2. Using List Comprehension and `max()` Function:

A more Pythonic way to achieve this is by using a list comprehension to extract the desired column into a new list, and then using the built-in `max()` function to get the largest value.

def find_max_in_column_comp(array_2d, column_index):
  if not array_2d or column_index < 0 or column_index >= len(array_2d[0]):
    return None # Return None for invalid input
  column_values = [row[column_index] for row in array_2d]
  return max(column_values)

# Example usage:
array = [[10, 20, 30], [40, 50, 60], [70, 80, 90]]
column_to_check = 2
max_in_column = find_max_in_column_comp(array, column_to_check)
print(f"Maximum value in column {column_to_check}: {max_in_column}") # Output: Maximum value in column 2: 90

Explanation:

- Like the previous method, it checks for invalid inputs and returns `None`.

- It uses list comprehension `[row[column_index] for row in array_2d]` to extract the values of the specified column from all rows into the list `column_values`.

- The built-in `max()` function is applied to the `column_values` list to retrieve the largest element in that column, which is then returned.

Both methods provide ways to retrieve the maximum value from a specific column in a 2D array, but the second approach with list comprehension is often preferred for its conciseness and readability.

Learn more about this