Site icon 지피디아

[python] 데이터를 다양한 형식으로 변경할 수 있는 to_로 시작하는 메서드

[python] 데이터를 다양한 형식으로 변경할 수 있는 to_로 시작하는 메서드

파이썬을 이용하다보면, 현재 불러온 데이터를 정제하여 다른 데이터 형식이나 확장자로 저장할 일이 있습니다. 이때 바로바로 사용할 수 있는 메서드들이 있습니다. to_로 시작하는 메서드들은 to_뒤에 무엇을 쓰느냐에 따라 원하는 모양으로 변경할 수 있습니다.

to_string

import pandas as pd

data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# to_string() 메서드를 사용하여 데이터프레임을 문자열로 변환
string_output = df.to_string(index=False, header=False)  # 인덱스와 열 이름을 출력하지 않음

print("=== Output ===")
print(string_output)
=== Output ===
 Alice   25  New York
   Bob   30     Paris
Charlie  35    London
  David   40     Tokyo

to_csv

import pandas as pd

data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# to_csv() 메서드를 사용하여 데이터프레임을 CSV 파일로 저장
df.to_csv('output.csv', sep='|', index=False)

print("=== Output ===")
print("CSV file 'output.csv' has been saved.")
=== Output ===
CSV file 'output.csv' has been saved.

to_excel

import pandas as pd

data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# to_excel() 메서드를 사용하여 데이터프레임을 Excel 파일로 저장
df.to_excel('output.xlsx', sheet_name='Sheet1', index=False)

print("=== Output ===")
print("Excel file 'output.xlsx' has been saved.")
=== Output ===
Excel file 'output.xlsx' has been saved.

to_html

import pandas as pd

data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# to_html() 메서드를 사용하여 데이터프레임을 HTML 형식으로 변환
html_output = df.to_html(index=False)

print("=== Output ===")
print(html_output)
=== Output ===
<table border="1" class="dataframe">
  <thead>
    <tr style="text-align: right;">
      <th>Name</th>
      <th>Age</th>
      <th>City</th>
    </tr>
  </thead>
  <tbody>
    <tr>
      <td>Alice</td>
      <td>25</td>
      <td>New York</td>
    </tr>
    <tr>
      <td>Bob</td>
      <td>30</td>
      <td>Paris</td>
    </tr>
    <tr>
      <td>Charlie</td>
      <td>35</td>
      <td>London</td>
    </tr>
    <tr>
      <td>David</td>
      <td>40</td>
      <td>Tokyo</td>
    </tr>
  </tbody>
</table>

to_dict

import pandas as pd

data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# to_dict() 메서드를 사용하여 데이터프레임을 딕셔너리로 변환
dict_output = df.to_dict(orient='records')  # records 형식으로 변환

print("=== Output ===")
print(dict_output)
=== Output ===
[{'Name': 'Alice', 'Age': 25, 'City': 'New York'}, {'Name': 'Bob', 'Age': 30, 'City': 'Paris'}, {'Name': 'Charlie', 'Age': 35, 'City': 'London'}, {'Name': 'David', 'Age': 40, 'City': 'Tokyo'}]

to_json

import json

data = {
    'name': 'John',
    'age': 30,
    'city': 'New York'
}

json_data = json.dumps(data)

print(json_data)
{"name": "John", "age": 30, "city": "New York"}

to_markdown

def to_markdown(data):
    markdown = ""
    for key, value in data.items():
        markdown += f"| {key} | {value} |\n"
    
    return markdown

data = {
    'name': 'John',
    'age': 30,
    'city': 'New York'
}

markdown_data = to_markdown(data)

print(markdown_data)
| name | John |
| age | 30 |
| city | New York |

to_clipboard

import pyperclip

data = {
    'name': 'John',
    'age': 30,
    'city': 'New York'
}

pyperclip.copy(str(data))
(클립보드에 복사됨)

to_sql

import sqlite3

def to_sql(data, table_name):
    conn = sqlite3.connect('data.db')
    cursor = conn.cursor()
    
    create_table_query = f"CREATE TABLE IF NOT EXISTS {table_name} (id INTEGER PRIMARY KEY, name TEXT, age INTEGER, city TEXT);"
    cursor.execute(create_table_query)
    
    insert_data_query = f"INSERT INTO {table_name} (name, age, city) VALUES (?, ?, ?);"
    cursor.execute(insert_data_query, (data['name'], data['age'], data['city']))
    
    conn.commit()
    conn.close()

data = {
    'name': 'John',
    'age': 30,
    'city': 'New York'
}

to_sql(data, 'person')
(SQLite 데이터베이스에 데이터가 저장됨)
Exit mobile version