CODEWITHSUNDEEP
pythonarrays
Ronaldinho Learn Coding
Ronaldinho Learn Coding

Reputation: 13824

Find length of 2D array Python

How do I find how many rows and columns are in a 2d array?

For example,

Input = ([[1, 2], [3, 4], [5, 6]])`

should be displayed as 3 rows and 2 columns.

Upvotes: 100

Views: 471788

Answers (6)

Miae Kim
Miae Kim

Reputation: 1771

assuming input[row][col]

rows = len(input)
cols = len(list(zip(*input)))

Upvotes: 0

Riya Das
Riya Das

Reputation: 11

You can also use np.size(a,1), 1 here is the axis and this will give you the number of columns

Upvotes: 1

MattNo
MattNo

Reputation: 291

In addition, correct way to count total item number would be:

sum(len(x) for x in input)

Upvotes: 28

Akavall
Akavall

Reputation: 86286

You can use numpy.shape.

import numpy as np
x = np.array([[1, 2],[3, 4],[5, 6]])

Result:

>>> x
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> np.shape(x)
(3, 2)

First value in the tuple is number rows = 3; second value in the tuple is number of columns = 2.

Upvotes: 43

machow
machow

Reputation: 1072

Assuming input[row][col],

    rows = len(input)
    cols = map(len, input)  #list of column lengths

Upvotes: 11

Óscar López
Óscar López

Reputation: 236124

Like this:

numrows = len(input)    # 3 rows in your example
numcols = len(input[0]) # 2 columns in your example

Assuming that all the sublists have the same length (that is, it's not a jagged array).

Upvotes: 183

Related Questions