Pandas 데이터프레임에서 열 수를 가져오는 방법
len() 함수 사용:
import pandas as pd
# 데이터프레임 생성
df = pd.DataFrame([[1, 2, 3], [4, 5, 6], [7, 8, 9]], columns=['A', 'B', 'C'])
# 열 수 출력
열_수 = len(df.columns)
print(열_수)
df.shape 속성 사용:
# 열 수 출력
열_수 = df.shape[1]
print(열_수)
df.columns 속성 사용:
# 열 수 출력
열_수 = len(df.columns)
print(열_수)
위의 방법 외에도 다음과 같은 방법들을 사용할 수 있습니다.
df.info()
함수 사용:
# 데이터프레임 정보 출력
df.info()
itertools.count()
함수 사용:
import itertools
# 열 수 출력
열_수 = sum(1 for _ in itertools.count(df.columns))
print(열_수)
참고:
- Pandas 데이터프레임에서 행 수를 가져오는 방법은 다음과 같습니다.
행_수 = len(df)
- 특정 열을 선택하려면 다음과 같이 사용할 수 있습니다.
특정_열 = df['열 이름']
import pandas as pd
# 데이터 생성
data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
'age': [30, 25, 22, 27],
'city': ['New York', 'Chicago', 'Denver', 'Los Angeles']}
# 데이터프레임 생성
df = pd.DataFrame(data)
# 방법 1: len() 함수 사용
열_수 = len(df.columns)
print("열 수 (len() 함수 사용):", 열_수)
# 방법 2: df.shape 속성 사용
열_수 = df.shape[1]
print("열 수 (df.shape 속성 사용):", 열_수)
# 방법 3: df.columns 속성 사용
열_수 = len(df.columns)
print("열 수 (df.columns 속성 사용):", 열_수)
열 수 (len() 함수 사용): 4
열 수 (df.shape 속성 사용): 4
열 수 (df.columns 속성 사용): 4
이 코드에서는 data
라는 딕셔너리를 사용하여 데이터프레임을 만듭니다. 딕셔너리에는 'name', 'age', 'city'라는 세 개의 키가 있으며 각 키는 값들의 리스트를 가지고 있습니다. 데이터프레임을 만든 후에는 len()
, df.shape
, df.columns
속성을 사용하여 열 수를 출력합니다. 세 가지 방법 모두 동일한 결과를 출력합니다.
추가 예제:
특정 열을 선택하려면 다음과 같이 사용할 수 있습니다.
# 'age' 열 선택
age_열 = df['age']
print(age_열)
Pandas 데이터프레임에서 열 수를 가져오는 대체 방법
itertools.islice() 함수 사용:
import pandas as pd
import itertools
# 데이터 생성
data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
'age': [30, 25, 22, 27],
'city': ['New York', 'Chicago', 'Denver', 'Los Angeles']}
# 데이터프레임 생성
df = pd.DataFrame(data)
# 열 수 출력
열_수 = sum(1 for _ in itertools.islice(df.columns, 1))
print("열 수 (itertools.islice() 함수 사용):", 열_수)
lambda 함수 사용:
# 열 수 출력
열_수 = len(list(filter(lambda x: True, df.columns)))
print("열 수 (lambda 함수 사용):", 열_수)
numpy.array() 함수 사용:
import numpy as np
# 열 수 출력
열_수 = len(np.array(df.columns))
print("열 수 (numpy.array() 함수 사용):", 열_수)
for 루프 사용:
# 열 수 출력
열_수 = 0
for _ in df.columns:
열_수 += 1
print("열 수 (for 루프 사용):", 열_수)
위 코드는 모두 동일한 결과를 출력합니다. 코드 선택은 개인 취향과 상황에 따라 다릅니다.
- 위의 대체 방법들은 일반적으로 기본적인 방법들보다 느릴 수 있습니다.
- 특정 상황에서 더 효율적인 방법이 있을 수도 있습니다.
python pandas dataframe