python 구구단 dataframe 저장

import pandas as pd

# 컬럼 이름과 인덱스 이름을 지정
# columns = [f'{i}단' for i in range(2, 10)]
columns = []
for i in range(2,10):
    columns.append(f'{i}단')
# index = [f'row{j}' for j in range(1, 10)]

# 데이터 생성
data = []

for j in range(1, 10):
    row = []
    for i in range(2, 10):
        row.append(f'{i} * {j} = {i*j}')
    data.append(row)

# DataFrame 생성
# df = pd.DataFrame(data, columns=columns, index=index)
df = pd.DataFrame(data, columns=columns)

# DataFrame을 엑셀 파일로 저장
df.to_excel('gugudan.xlsx', index=False)

# 저장된 엑셀 파일 확인
print('gugudan.xlsx 파일 저장 완료!')

 

import pandas as pd
import numpy as np

def create_gugudan_df():
    # 2단부터 9단까지의 컬럼 생성
    columns = [f'{i}단' for i in range(2, 10)]
    
    # 1부터 9까지 곱하는 수 생성
    multipliers = np.arange(1, 10)
    
    # 구구단 계산을 리스트 컴프리헨션으로 생성
    data = [
        [f'{col} × {row} = {col*row}' 
         for col in range(2, 10)]
        for row in multipliers
    ]
    
    # DataFrame 생성 및 반환
    return pd.DataFrame(data, columns=columns)

def save_gugudan_excel(filename='gugudan.xlsx'):
    df = create_gugudan_df()
    df.to_excel(filename, index=False)
    print(f'{filename} 파일이 성공적으로 저장되었습니다!')

if __name__ == '__main__':
    save_gugudan_excel()