Матрица из биграммы? - PullRequest
Новая
       65

Матрица из биграммы?

2 голосов
/ 12 июля 2020

У меня есть список аккордов 

['F', 'Em7', 'A7', 'Dm', 'Dm7', 'A#', 'C7', 'C', 'G7', 'A7sus4', 'Gm6', 'Fsus4', etc]

, и я хочу превратить его в матрицу переходов. Что у меня есть: 

import pandas as pd
from nltk.util import ngrams

bigram = pd.Series(ngrams(data.split(),2)) ### creating the bigrams of ordered pairs like:
### 0      (F, Em7) 
### 1     (Em7, A7) 
### 2      (A7, Dm)

probabilities = bigram.value_counts(normalize=True) ### getting probability of each ordered pair
letters = pd.Series(data.split()).unique() ### getting each chord

Я хочу, чтобы матрица была такой: 

          F        Em7      A7          ....
 F     p(FF)    p(FEm7)    p(FA7)       ....
 Em7   p(Em7F)  p(Em7Em7)  p(Em7A7)     ....
 A7    p(A7F)    p(A7Em7)  p(A7A7)      ....
  .
  .
  .

, где p(FF) - это вероятность (F, F), которая находится в моем списке вероятностей. Как я могу это сделать? Спасибо !!

1 Ответ

0 голосов
/ 12 июля 2020

Создайте новый pd.Series с индексом MultiLevel, созданным с использованием упорядоченных пар bigrams из списка prob, затем используйте Series.unstack с fill_value=0, чтобы преобразовать ряд в сводные данные, опционально round значения: 

prob = bigram.value_counts(normalize=True)

mat = (
    pd.Series(
        prob, index=[prob.index.str[0], prob.index.str[1]])
    .unstack(fill_value=0).round(3)
)

Результат: 

print(mat)

           A#     A7  A7sus4      C     C7     Dm    Dm7    Em7  Fsus4     G7    Gm6
A#      0.000  0.000   0.000  0.000  0.091  0.000  0.000  0.000  0.000  0.000  0.000
A7      0.000  0.000   0.000  0.000  0.000  0.091  0.000  0.000  0.000  0.000  0.000
A7sus4  0.000  0.000   0.000  0.000  0.000  0.000  0.000  0.000  0.000  0.000  0.091
C       0.000  0.000   0.000  0.000  0.000  0.000  0.000  0.000  0.000  0.091  0.000
C7      0.000  0.000   0.000  0.091  0.000  0.000  0.000  0.000  0.000  0.000  0.000
Dm      0.000  0.000   0.000  0.000  0.000  0.000  0.091  0.000  0.000  0.000  0.000
Dm7     0.091  0.000   0.000  0.000  0.000  0.000  0.000  0.000  0.000  0.000  0.000
Em7     0.000  0.091   0.000  0.000  0.000  0.000  0.000  0.000  0.000  0.000  0.000
F       0.000  0.000   0.000  0.000  0.000  0.000  0.000  0.091  0.000  0.000  0.000
G7      0.000  0.000   0.091  0.000  0.000  0.000  0.000  0.000  0.000  0.000  0.000
Gm6     0.000  0.000   0.000  0.000  0.000  0.000  0.000  0.000  0.091  0.000  0.000
...