Предположим, у меня есть датафрейм pandas с несколькими столбцами, например:

import pandas as pd
iterables = [['a', 'b'], ['1', '2']]
my_index = pd.MultiIndex.from_product(iterables, names=['first', 'second'])
df = pd.DataFrame([[1, 2, 3, 4], [5, 6, 7, 8]], columns=my_index)

Тогда df производит

first  a   b
second 1 2 1 2
0      1 2 3 4
1      5 6 7 8

Теперь, если я хочу самокорреляции df['a'] с самим собой, это просто: df['a'].corr() поймет меня. Обратите внимание, что такая корреляция имеет форму (2, 2).

Я хотел бы рассчитать матрицу корреляции df['a'] с df['b']. Предположительно, код df['a'].corrwith(df['b']) должен дать мне это. Этот код выполняется, но результат имеет форму (2,), которая мне не подходит. Почему матрица самокорреляции, заданная .corr(), должна давать результат, отличный от корреляции, заданной .corrwith()? Мне нужна корреляционная матрица той же формы, что и df['a'].corr(), потому что я хочу построить тепловые карты Seaborn и мне нужна двумерная корреляционная матрица.

Спасибо заранее за ваше время!