Изменение значений столбца в заголовках столбцов в pandas

У меня есть следующий код, который принимает значения в одном столбце блока данных pandas и делает их столбцами нового фрейма данных. Значения в первом столбце блока данных становятся индексом нового кадра данных.

В некотором смысле, я хочу превратить список смежности в матрицу смежности. Вот код:

• HTML не корректно отображается с Canopy 1.7.1.3323 / IPython 4.1.2
• Python Pandas: как заменить символы в столбце фрейма данных?
• Сравнение строк в таблице оракула и сопоставление обновлений
• Доступ к оси или фигуре в python ggplot
• Как вычислить сходство jaccard с рамкой данных pandas

import pandas as pa print "Original Data Frame" # Create a dataframe oldcols = {'col1':['a','a','b','b'], 'col2':['c','d','c','d'], 'col3':[1,2,3,4]} a = pa.DataFrame(oldcols) print a # The columns of the new data frame will be the values in col2 of the original newcols = list(set(oldcols['col2'])) rows = list(set(oldcols['col1'])) # Create the new data matrix data = np.zeros((len(rows), len(newcols))) # Iterate over each row and fill in the new matrix for row in zip(a['col1'], a['col2'], a['col3']): rowindex = rows.index(row[0]) colindex = newcols.index(row[1]) data[rowindex][colindex] = row[2] newf = pa.DataFrame(data) newf.columns = newcols newf.index = rows print "New data frame" print newf 

Это работает для этого конкретного примера:

 Original Data Frame col1 col2 col3 0 ac 1 1 ad 2 2 bc 3 3 bd 4 New data frame cd a 1 2 b 3 4 

Он будет терпеть неудачу, если значения в col3 не являются числами. Мой вопрос в том, есть ли более элегантный / надежный способ сделать это?

 import pandas as pd oldcols = {'col1':['a','a','b','b'], 'col2':['c','d','c','d'], 'col3':[1,2,3,4]} a = pd.DataFrame(oldcols) newf = a.pivot(index='col1', columns='col2') print(newf) 

  col3 col2 cd col1 a 1 2 b 3 4 

 newf.columns = newf.columns.droplevel(0) 

 col2 cd col1 a 1 2 b 3 4