У меня есть пустой массив, как показано ниже.
value array:
[[ 5 58 7 22 8 50]
[ 7 22 9 22 10 12]
[ 1 44 4 12 8 50]
[ 1 4 4 2 8 2]
[ 7 17 8 17 10 7]
[ 7 17 9 17 10 7]
[ 8 17 9 17 10 7]
[ 8 1 9 1 10 1]
[ 8 2 9 7 10 37]]
Я использую этот код, чтобы получить четные столбцы значений и сгруппировать его, как это:
k=3
index_feature = list(range(0, 2 * k, 2))
group_feature = npi.group_by(value_array[:,index_feature]).split(value_array[:, index_feature])
group_feature: (массив ([[1, 4, 8]]), массив ([[5, 7, 8]]), массив ([[7, 8, 10]]), массив ([[7, 9, 10]]), массив ([[8, 9, 10]]))
А теперь я хочу фильтровать следующим образом: если столбец 0, 2, 4 равен 1, 4, 8, который [1, 4, 8] является элементом в group_feature, и получить результаты:
[[4,2,2]]
Вероятно, если массив ([[8, 9, 10]])), я получу:
[[17, 17, 7]
[1, 1, 1]
[2, 7, 37]]
Как я могу это сделать?