- 方法1: 2重ループを使用する この方法では、行列の各要素の組み合わせをチェックし、目的のペアを見つけます。
def find_pair(matrix, target):
for i in range(len(matrix)):
for j in range(len(matrix[0])):
if matrix[i][j] == target:
return i, j
return -1, -1
# 使用例:
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
target = 5
pair = find_pair(matrix, target)
if pair != (-1, -1):
print(f"目的のペア ({target}) は行 {pair[0]} 列 {pair[1]} にあります。")
else:
print("目的のペアは見つかりませんでした。")
- 方法2: NumPyを使用する NumPyは、行列操作に便利な機能を提供しています。次のコード例では、NumPyを使用して目的のペアを見つける方法を示します。
import numpy as np
def find_pair(matrix, target):
indices = np.where(matrix == target)
if len(indices[0]) > 0:
return indices[0][0], indices[1][0]
return -1, -1
# 使用例:
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
target = 5
pair = find_pair(matrix, target)
if pair != (-1, -1):
print(f"目的のペア ({target}) は行 {pair[0]} 列 {pair[1]} にあります。")
else:
print("目的のペアは見つかりませんでした。")
これらの方法を使用すると、行列内の特定のペアを見つけることができます。上記のコード例を参考にして、自分のデータに適用してみてください。