Вращение матрицы с несколькими значениями - PullRequest
Новая
       35

Вращение матрицы с несколькими значениями

0 голосов
/ 06 декабря 2018

Я использую Python 3.7.1.

Учитывая следующий одномерный массив (чисел в диапазоне от -1000 до 1000): 

[x00,y00,z00, x10,y10,z10, x20,y20,z20, x30,y30,z30,
 x01,y01,z01, x11,y11,z11, x21,y21,z21, x31,y31,z31,
 x02,y02,z02, x12,y12,z12, x22,y22,z22, x32,y32,z32]

Я хочу использовать это вращениематрица для вращения. 

|0 -1|
|1  0|

Требуемый вывод: 

[x30,y30,z30, x31,y31,z31, x32,y32,z32,
 x20,y20,z20, x21,y21,z21, x22,y22,z22,
 x10,y10,z10, x11,y11,z11, x12,y12,z12,
 x00,y00,z00, x01,y01,z01, x02,y02,z02]

Я знаю, как это можно сделать для обычного массива, но я хочу, чтобы значения x, y и z были сгруппированы.

1 Ответ

0 голосов
/ 06 декабря 2018

Я смог сделать это с помощью предложения @DatHydroGuy о numpy.rot90.Вот примеры того, как это сделать.Обратите внимание, что сначала я группирую значения x, y, z в кортежах в списке.Затем создайте пустой массив кортежей в виде объектов, поверните его, а затем сведите массив в список, используя понимание списка. 

import numpy as np

a = [5, 0, 3, 3, 7, 9, 3, 5, 2, 4, 7, 6, 8, 8, 1, 6, 7, 7, 8, 1, 5, 9, 8, 9, 4, 3, 0, 3, 5, 0, 2, 3, 8, 1, 3, 3]
sh = (4,3) # Shape

a_tup = [tuple(a[i:i+3]) for i in range(0, len(a), 3)] # Group list into tuples (x,y,z)
print(a_tup)
# [(5, 0, 3), (3, 7, 9), (3, 5, 2), (4, 7, 6), (8, 8, 1), (6, 7, 7), (8, 1, 5), (9, 8, 9), (4, 3, 0), (3, 5, 0), (2, 3, 8), (1, 3, 3)]
b=np.empty(sh, dtype=object) # Initialize numpy array with object as elements (for the tuples) and your shape sh
for j in range(sh[1]): # Assign tuples from list  to positions in the array
  for i in range(sh[0]):
    b[i,j] = a_tup[i+j]
print(b)
# [[(5, 0, 3) (3, 7, 9) (3, 5, 2)]
#  [(3, 7, 9) (3, 5, 2) (4, 7, 6)]
#  [(3, 5, 2) (4, 7, 6) (8, 8, 1)]
#  [(4, 7, 6) (8, 8, 1) (6, 7, 7)]]
c = np.rot90(b)
print(c)
# [[(3, 5, 2) (4, 7, 6) (8, 8, 1) (6, 7, 7)]
#  [(3, 7, 9) (3, 5, 2) (4, 7, 6) (8, 8, 1)]
#  [(5, 0, 3) (3, 7, 9) (3, 5, 2) (4, 7, 6)]]
print([item for sublist in c.flatten() for item in sublist]) # Flatten the numpy array of tuples to a list of numbers
# [3, 5, 2, 4, 7, 6, 8, 8, 1, 6, 7, 7, 3, 7, 9, 3, 5, 2, 4, 7, 6, 8, 8, 1, 5, 0, 3, 3, 7,9, 3, 5, 2, 4, 7, 6]
...