How to access a column in a list of lists in python

by

List comprehensions are your friend when working with lists of lists:

In [111]: alist
Out[111]: 
[[0, 1, 2, 3, 4, 5],
 [6, 7, 8, 9, 10, 11],
 [12, 13, 14, 15, 16, 17],
 [18, 19, 20, 21, 22, 23]]
In [112]: [row[1] for row in alist]
Out[112]: [1, 7, 13, 19]

There’s also a handy ‘idiom’ for transposing a nested list, turning ‘columns’ into ‘rows’:

In [113]: tlist = list(zip(*alist))
In [114]: tlist
Out[114]: 
[(0, 6, 12, 18),
 (1, 7, 13, 19),
 (2, 8, 14, 20),
 (3, 9, 15, 21),
 (4, 10, 16, 22),
 (5, 11, 17, 23)]
In [115]: tlist[1]
Out[115]: (1, 7, 13, 19)

More Related Contents:

  1. What are the advantages of NumPy over regular Python lists?
  2. Find nearest indices for one array against all values in another array – Python / NumPy
  3. List to array conversion to use ravel() function
  4. Numpy individual element access slower than for lists
  5. How to create a list of list from the given input? [closed]
  6. Find the row indexes of several values in a numpy array
  7. What are the differences between numpy arrays and matrices? Which one should I use?
  8. How to read a text file into a list or an array with Python
  9. Working with big data in python and numpy, not enough ram, how to save partial results on disc?
  10. How to apply a disc shaped mask to a NumPy array?
  11. How to remove specific elements in a numpy array
  12. Get the position of the largest value in a multi-dimensional NumPy array
  13. Using Numpy Vectorize on Functions that Return Vectors
  14. How to extend an array in-place in Numpy?
  15. np arrays being immutable – “assignment destination is read-only”
  16. How to turn a video into numpy array?
  17. Can numpy bincount work with 2D arrays?
  18. Sort invariant for numpy.argsort with multiple dimensions
  19. Numpy isnan() fails on an array of floats (from pandas dataframe apply)
  20. Concatenate range arrays given start, stop numbers in a vectorized way – NumPy
  21. Numpy – Replace a number with NaN
  22. Numpy select rows based on condition
  23. Python recursive function to display all subsets of given set
  24. Sorting a python array/recarray by column
  25. Numpy Array Get row index searching by a row
  26. Unexpected Behavior of Extend with a list in Python [duplicate]
  27. Resample a numpy array
  28. Creating a numpy array of 3D coordinates from three 1D arrays
  29. How to use Python multiprocessing Pool.map to fill numpy array in a for loop
  30. Is this the best way to add an extra dimension to a numpy array in one line of code?

Leave a Comment