[python] 팬더 데이터 프레임에서 행의 하위 집합 수정

A와 B라는 두 개의 열이있는 팬더 DataFrame이 있다고 가정합니다. A가 0 일 때마다 B가 항상 NaN이되도록이 DataFrame을 수정하거나 복사본을 만들고 싶습니다. 어떻게하면 되나요?

나는 다음을 시도했다

df['A'==0]['B'] = np.nan

df['A'==0]['B'].values.fill(np.nan)

성공없이.



답변

사용 .loc레이블 기반 인덱싱 :

df.loc[df.A==0, 'B'] = np.nan

df.A==0표현식은 행을 색인화하고 열을 'B'선택 하는 부울 계열을 작성합니다 . 이를 사용하여 열의 하위 집합을 변환 할 수도 있습니다. 예 :

df.loc[df.A==0, 'B'] = df.loc[df.A==0, 'B'] / 2

나는 그것이 왜 작동하는지 정확히 알기 위해 팬더 내부에 대해 충분히 알지 못하지만 기본 문제는 때때로 DataFrame으로 인덱싱하면 결과의 사본을 반환하고 때로는 원래 객체에 대한보기를 반환한다는 것입니다. here 문서에 따르면 이 동작은 기본 numpy 동작에 따라 다릅니다. 한 작업 ([1] [2] 대신)으로 모든 것에 액세스하는 것이 설정 작업에 더 효과적이라는 것을 알았습니다.


답변

고급 인덱싱에 대한 팬더 문서는 다음과 같습니다 .

이 섹션에서는 필요한 것을 정확하게 설명합니다! 알고 보니 df.loc(.ix가 사용되지 않습니다으로 – 아래 지적한 많은 같은) dataframe의 다이 싱 / 쿨 슬라이스 사용할 수 있습니다. 과. 또한 물건을 설정하는 데 사용될 수 있습니다.

df.loc[selection criteria, columns I want] = value

브렌의 대답은 말하고 그래서 ‘나에게 모든 장소 찾아 df.A == 0선택 열을 B하고로 설정 np.nan


답변

팬더 0.20 ix 부터는 더 이상 사용되지 않습니다 . 올바른 방법은 df.loc 를 사용하는 것입니다

다음은 실제 예입니다

>>> import pandas as pd 
>>> import numpy as np 
>>> df = pd.DataFrame({"A":[0,1,0], "B":[2,0,5]}, columns=list('AB'))
>>> df.loc[df.A == 0, 'B'] = np.nan
>>> df
   A   B
0  0 NaN
1  1   0
2  0 NaN
>>> 

설명:

여기 문서 에서 설명했듯이 , .loc 주로 레이블 기반이지만 부울 배열과 함께 사용될 수도 있습니다 .

따라서 우리가 위에서하고있는 일은 df.loc[row_index, column_index]다음에 의해 적용 됩니다.

  • loc부울 배열을 마스크로 사용하여 팬더에게 변경하려는 행의 하위 집합을 알려주 는 사실을 악용row_index
  • 사실을 악용하는 것은 loc또한 라벨을 사용하여 열을 선택하는 기준으로 레이블입니다 'B'의를column_index

논리, 조건 또는 일련의 부울을 반환하는 모든 연산을 사용하여 부울 배열을 구성 할 수 있습니다. 위의 예에서 우리는 어떤 원하는 rows를 포함하는 0우리가 사용할 수에 대한, df.A == 0아래의 예를 불리언이 반환 시리즈에서 볼 수 있듯이을.

>>> df = pd.DataFrame({"A":[0,1,0], "B":[2,0,5]}, columns=list('AB'))
>>> df
   A  B
0  0  2
1  1  0
2  0  5
>>> df.A == 0
0     True
1    False
2     True
Name: A, dtype: bool
>>> 

그런 다음 위의 부울 배열을 사용하여 필요한 행을 선택하고 수정합니다.

>>> df.loc[df.A == 0, 'B'] = np.nan
>>> df
   A   B
0  0 NaN
1  1   0
2  0 NaN

자세한 내용은 여기 에서 고급 인덱싱 설명서를 확인 하십시오 .


답변

속도를 크게 높이려면 NumPy의 where 함수를 사용하십시오.

설정

행이 100,000 인 행이 2 개인 DataFrame을 0으로 만듭니다.

df = pd.DataFrame(np.random.randint(0,3, (100000,2)), columns=list('ab'))

빠른 솔루션 numpy.where

df['b'] = np.where(df.a.values == 0, np.nan, df.b.values)

타이밍

%timeit df['b'] = np.where(df.a.values == 0, np.nan, df.b.values)
685 µs ± 6.4 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

%timeit df.loc[df['a'] == 0, 'b'] = np.nan
3.11 ms ± 17.2 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

Numpy where는 약 4 배 빠릅니다.


답변

여러 열을 바꾸려면 .values다음을 사용하여 numpy 배열로 변환하십시오 .

df.loc[df.A==0, ['B', 'C']] = df.loc[df.A==0, ['B', 'C']].values / 2


답변