(김*준) (강의노트15장) .squeeze()관련

[chobl0002]

위 사진에서 .squeeze()를 사용한 것과 사용하지 않은 것의 표 모양은 같아 보이는데, .squeeze()를 사용했을 때의 이점이 무엇인지 여쭙고 싶습니다.

[Kdy0412]

• squeeze()는 차원 중 길이가 1인 축을 제거하는 메서드입니다.

squeeze()사용의 예시 코드

import numpy as np
import pandas as pd
data_url = "https://raw.githubusercontent.com/codingalzi/code-workout-datasci/refs/heads/master/data/"
fathers_2 = pd.read_csv(data_url+"pearson_dataset.csv", usecols = [0])
fathers_1 = pd.read_csv(data_url+"pearson_dataset.csv", usecols = [0]).squeeze()  #squeeze()메서드 사용. 불필요한 차원을 줄인다.
print(fathers_2.shape,fathers_1.shape)

(1078, 1) (1078,)

squeeze() 메서드로 인해 차원이 줄어든 모습입니다.
물론 그냥 출력하면 표의 형태는 같아 보이지만 fathers_2와 fathers_1는 차원이 다릅니다. 즉 fathers_2는 데이터프레임이고 fathers_1은 시리즈입니다.

type(fathers_2)
type(fathers_1)

pandas.core.frame.DataFrame
pandas.core.series.Series

• 데이터프레임이란?

따라서 squeeze()메서드 사용의 이점이라 하면 불필요한 차원 ex) 길이가 1인 축을 제거 2차원 DataFrame -> 1차원 Series함으로써 데이터를 더 간단하고 다루기 쉬운 형태로 만드는 것이라고 생각합니다.

• 외부링크: https://numpy.org/doc/2.2/reference/generated/numpy.squeeze.html#numpy.squeeze