import numpy as np

class Solution:
    def findColumnWidth(self, grid: List[List[int]]) -> List[int]:
         # let's convert the grid to a numpy array
        np_grid = np.array(grid, dtype=str)
        # calculate the length of each element in the grid
        lengths = np.vectorize(len)(np_grid)
        # find the maximum length of each column
        max_lengths = lengths.max(axis=0)
        return max_lengths.tolist()

impl Solution {
    pub fn find_column_width(grid: Vec<Vec<i32>>) -> Vec<i32> {
        let col_n = grid[0].len();
        let mut ans = vec![0; col_n];

        for row in grid.iter() {
            for (i, &num) in row.iter().enumerate() {
                let length = num.to_string().len() as i32;
                if length > ans[i] {
                    ans[i] = length;
                }
            }
        }
        ans
    }
}