Транспонирование матрицы в Python
Я пытаюсь создать функцию транспонирования матрицы для python, но я не могу заставить ее работать.
Скажем, у меня есть
theArray = [['a','b','c'],['d','e','f'],['g','h','i']]
и я хочу, чтобы моя функция, чтобы придумать
newArray = [['a','d','g'],['b','e','h'],['c', 'f', 'i']]
другими словами, если бы я напечатал этот 2D-массив в виде столбцов и строк, я бы хотел, чтобы строки превратились в столбцы и столбцы в строки.
Я сделал это до сих пор, но это не работает
def matrixTranspose(anArray):
transposed = [None]*len(anArray[0])
for t in range(len(anArray)):
for tt in range(len(anArray[t])):
transposed[t] = [None]*len(anArray)
transposed[t][tt] = anArray[tt][t]
print transposed
15 ответов:
Python 2:
>>> theArray = [['a','b','c'],['d','e','f'],['g','h','i']] >>> zip(*theArray) [('a', 'd', 'g'), ('b', 'e', 'h'), ('c', 'f', 'i')]Python 3:
>>> [*zip(*theArray)] [('a', 'd', 'g'), ('b', 'e', 'h'), ('c', 'f', 'i')]
>>> theArray = [['a','b','c'],['d','e','f'],['g','h','i']] >>> [list(i) for i in zip(*theArray)] [['a', 'd', 'g'], ['b', 'e', 'h'], ['c', 'f', 'i']]генератор списков создает новый 2d массив с элементами списка вместо кортежей.
если строки не равны, вы также можете использовать
map:>>> uneven = [['a','b','c'],['d','e'],['g','h','i']] >>> map(None,*uneven) [('a', 'd', 'g'), ('b', 'e', 'h'), ('c', None, 'i')]Edit: в Python 3 функциональность
mapизменен,itertools.zip_longestможно использовать вместо:
Источник:Что нового в Python 3.0>>> import itertools >>> uneven = [['a','b','c'],['d','e'],['g','h','i']] >>> list(itertools.zip_longest(*uneven)) [('a', 'd', 'g'), ('b', 'e', 'h'), ('c', None, 'i')]
гораздо проще с numpy:
>>> arr = np.array([[1,2,3],[4,5,6],[7,8,9]]) >>> arr array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) >>> arr.T array([[1, 4, 7], [2, 5, 8], [3, 6, 9]]) >>> theArray = np.array([['a','b','c'],['d','e','f'],['g','h','i']]) >>> theArray array([['a', 'b', 'c'], ['d', 'e', 'f'], ['g', 'h', 'i']], dtype='|S1') >>> theArray.T array([['a', 'd', 'g'], ['b', 'e', 'h'], ['c', 'f', 'i']], dtype='|S1')
проблема с вашим исходным кодом была в том, что вы инициализировали
transpose[t]в каждом элементе, а не только один раз в строке:def matrixTranspose(anArray): transposed = [None]*len(anArray[0]) for t in range(len(anArray)): transposed[t] = [None]*len(anArray) for tt in range(len(anArray[t])): transposed[t][tt] = anArray[tt][t] print transposedэто работает, хотя есть более подходящие для Python способов сделать те же самые вещи, в том числе @Джей Ф.
zipприложение.
чтобы завершить ответ J. F. Sebastian, если у вас есть список списков с разной длиной, проверьте это отличный пост от ActiveState. Короче говоря:
встроенная функция zip выполняет аналогичную работу, но усекает результат к длине самого короткого списка, поэтому некоторые элементы из оригинала данные могут быть потеряны.
для обработки списка списков разной длины используйте:
def transposed(lists): if not lists: return [] return map(lambda *row: list(row), *lists) def transposed2(lists, defval=0): if not lists: return [] return map(lambda *row: [elem or defval for elem in row], *lists)
" лучший " ответ уже был представлен, но я подумал, что добавлю, что вы можете использовать вложенные списки понимания, как видно в Python Tutorial.
вот как вы могли бы получить транспонированный массив:
def matrixTranspose( matrix ): if not matrix: return [] return [ [ row[ i ] for row in matrix ] for i in range( len( matrix[ 0 ] ) ) ]
этот сохранит прямоугольную форму, так что последующие транспозиции получат правильный результат:
import itertools def transpose(list_of_lists): return list(itertools.izip_longest(*list_of_lists,fillvalue=' '))
def matrixTranspose(anArray): transposed = [None]*len(anArray[0]) for i in range(len(transposed)): transposed[i] = [None]*len(transposed) for t in range(len(anArray)): for tt in range(len(anArray[t])): transposed[t][tt] = anArray[tt][t] return transposed theArray = [['a','b','c'],['d','e','f'],['g','h','i']] print matrixTranspose(theArray)
#generate matrix matrix=[] m=input('enter number of rows, m = ') n=input('enter number of columns, n = ') for i in range(m): matrix.append([]) for j in range(n): elem=input('enter element: ') matrix[i].append(elem) #print matrix for i in range(m): for j in range(n): print matrix[i][j], print '\n' #generate transpose transpose=[] for j in range(n): transpose.append([]) for i in range (m): ent=matrix[i][j] transpose[j].append(ent) #print transpose for i in range (n): for j in range (m): print transpose[i][j], print '\n'
a=[] def showmatrix (a,m,n): for i in range (m): for j in range (n): k=int(input("enter the number") a.append(k) print (a[i][j]), print('\t') def showtranspose(a,m,n): for j in range(n): for i in range(m): print(a[i][j]), print('\t') a=((89,45,50),(130,120,40),(69,79,57),(78,4,8)) print("given matrix of order 4x3 is :") showmatrix(a,4,3) print("Transpose matrix is:") showtranspose(a,4,3)
def transpose(matrix): x=0 trans=[] b=len(matrix[0]) while b!=0: trans.append([]) b-=1 for list in matrix: for element in list: trans[x].append(element) x+=1 x=0 return trans
def transpose(matrix): listOfLists = [] for row in range(len(matrix[0])): colList = [] for col in range(len(matrix)): colList.append(matrix[col][row]) listOfLists.append(colList) return listOfLists
программа Python для транспонирования матрицы:
row,col = map(int,input().split()) matrix = list() for i in range(row): r = list(map(int,input().split())) matrix.append(r) trans = [[0 for y in range(row)]for x in range(col)] for i in range(len(matrix[0])): for j in range(len(matrix)): trans[i][j] = matrix[j][i] for i in range(len(trans)): for j in range(len(trans[0])): print(trans[i][j],end=' ') print(' ')
Comments