01 May 2022

How to slice a number of columns from a 2-dimensional list in Python?

https://www.codegrepper.com/code-examples/python/python+multiaxis+slicing

# Basic syntax:
[your_list[i][start_col : end_col] for i in range(start_row, end_row)]

# Example usage:
your_list = [[1, 2, 3, 4, 5],[6, 7, 8, 9, 10],[11, 12, 13, 14, 15]]
# obtain rows 0-1 (inclusive) and cols 2-3 (inclusive)
[your_list[i][2:4] for i in range(0,2)]
--> [[3, 4], [8, 9]]

No comments:

Post a Comment