Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Chương trình Python trả về hàng có phần tử tại chỉ số xác định

So sánh hai ma trận cùng kích thước, trả về hàng của ma trận đầu tiên nơi phần tử tại chỉ số key trùng khớp giữa hai ma trận.

Ví dụ

Minh họa:

my_list_1 = [[21, 81, 35], [91, 14, 0], [64, 61, 42]]
my_list_2 = [[21, 92, 63], [80, 19, 65], [54, 65, 36]]
print("Ma trận 1:")
print(my_list_1)
print("Ma trận 2:")
print(my_list_2)
my_key = 0
my_result = []
for index in range(len(my_list_1)):
    if my_list_1[index][my_key] == my_list_2[index][my_key]:
        my_result.append(my_list_1[index])
        my_result.append(my_list_1[index])  # append twice per original logic
print("Kết quả:")
print(my_result)

Kết quả

Ma trận 1: [[21, 81, 35], [91, 14, 0], [64, 61, 42]]
Ma trận 2: [[21, 92, 63], [80, 19, 65], [54, 65, 36]]
Kết quả: [[21, 81, 35], [21, 81, 35]]

Giải thích

  • Duyệt chỉ số hàng, so sánh my_list_1[i][key] == my_list_2[i][key].
  • Nếu khớp, thêm hàng của ma trận 1 vào kết quả (ví dụ thêm 2 lần theo code gốc).