Pandas merge를 사용할 때 색인을 유지하는 방법

두 개를 병합 DataFrames하고 첫 번째 프레임의 인덱스를 병합 된 데이터 세트의 인덱스로 유지하고 싶습니다 . 그러나 병합을 수행하면 결과 DataFrame에 정수 인덱스가 있습니다. 왼쪽 데이터 프레임에서 인덱스를 유지하도록 지정하려면 어떻게해야합니까?

In [4]: a = pd.DataFrame({'col1': {'a': 1, 'b': 2, 'c': 3}, 
                          'to_merge_on': {'a': 1, 'b': 3, 'c': 4}})

In [5]: b = pd.DataFrame({'col2': {0: 1, 1: 2, 2: 3}, 
                          'to_merge_on': {0: 1, 1: 3, 2: 5}})

In [6]: a
Out[6]:
   col1  to_merge_on
a     1            1
b     2            3
c     3            4

In [7]: b
Out[7]:
   col2  to_merge_on
0     1            1
1     2            3
2     3            5

In [8]: a.merge(b, how='left')
Out[8]:
   col1  to_merge_on  col2
0     1            1   1.0
1     2            3   2.0
2     3            4   NaN

In [9]: _.index
Out[9]: Int64Index([0, 1, 2], dtype='int64')

편집 : 쉽게 재현 할 수있는 예제 코드로 전환

In [5]: a.reset_index().merge(b, how="left").set_index('index')
Out[5]:
       col1  to_merge_on  col2
index
a         1            1     1
b         2            3     2
c         3            4   NaN

참고 : 사이에 여러 개의 일치가있는 경우 일부 왼쪽 병합 작업에 더 많은 행을 끝낼 수 a및 b당신은 중복 제거 (필요 중복 제거에 대한 문서가 ). 이것이 팬더가 색인을 유지하지 않는 이유입니다.

왼쪽 데이터 프레임에 인덱스 복사본을 만들고 병합 할 수 있습니다.

a['copy_index'] = a.index
a.merge(b, how='left')

이 간단한 방법은 큰 데이터 프레임으로 작업하고 pd.merge_asof()(또는 dd.merge_asof())을 사용하는 동안 매우 유용하다는 것을 알았습니다 .

이 접근 방식은 인덱스 재설정이 비용이 많이 드는 경우 (대용량 데이터 프레임) 더 우수합니다.

비 pd.merge 솔루션이 있습니다. map및 사용set_index

In [1744]: a.assign(col2=a['to_merge_on'].map(b.set_index('to_merge_on')['col2']))
Out[1744]:
   col1  to_merge_on  col2
a     1            1   1.0
b     2            3   2.0
c     3            4   NaN

그리고 index인덱스에 대한 더미 이름을 도입하지 않습니다 .

df1 = df1.merge(
        df2, how="inner", left_index=True, right_index=True
    )

이를 통해 df1의 색인을 유지할 수 있습니다.

내가 다른 해결책을 생각해 냈다고 생각합니다. 왼쪽 테이블의 인덱스를 기반으로 한 열 값에 인덱스 값과 오른쪽 테이블을 조인했습니다. 내가 한 일은 정상적인 병합이었습니다.

First10ReviewsJoined = pd.merge(First10Reviews, df, left_index=True, right_on='Line Number')

그런 다음 병합 된 테이블에서 새 인덱스 번호를 검색하여 Sentiment Line Number라는 새 열에 넣습니다.

First10ReviewsJoined['Sentiment Line Number']= First10ReviewsJoined.index.tolist()

그런 다음 Line Number (왼쪽 테이블 인덱스에서 조인 한 열 값)라는 기존 열을 기반으로 인덱스를 원래의 왼쪽 테이블 인덱스로 다시 수동 설정합니다.

First10ReviewsJoined.set_index('Line Number', inplace=True)

그런 다음 행 번호의 색인 이름을 제거하여 공백으로 둡니다.

First10ReviewsJoined.index.name = None

아마도 약간의 해킹이지만 잘 작동하고 비교적 간단하게 보입니다. 또한 데이터가 중복되거나 엉망이 될 위험을 줄여줍니다. 모든 것이 이해되기를 바랍니다.

또 다른 간단한 옵션은 인덱스의 이름을 이전과 같이 바꾸는 것입니다.

a.merge(b, how="left").set_axis(a.index)

merge는 데이터 프레임 'a'에서 순서를 유지하지만 인덱스를 재설정하여 set_axis를 사용하도록 저장합니다.