Python中使用矩陣經常需要取部分元素,可以使用切片來實現。其中,切片的語法使用[start:end]的格式,通過指定start和end的值來控制提取的元素的范圍。在該格式中,start是元素開始位置的下標,end是元素結束位置的下標。
# 創建一個 2x2 的矩陣 matrix = [[1, 2], [3, 4]] # 取第一行的所有元素 print(matrix[0][:]) # [1, 2] # 取第二列的所有元素 print([row[1] for row in matrix]) # [2, 4] # 取左上角的元素 print(matrix[:1][:1]) # [[1, 2]] # 取右下角的元素 print(matrix[1:2][1:2]) # [[4]]
在使用切片的時候需要注意索引的下標范圍,否則可能會產生IndexError(下標越界)的錯誤。