pandasでCSVの生成と読み込み

2020年12月21日

pandasでCSVの生成と読み込み

データベースから読み込んだ値をデータフレーム化して、CSVの生成とCSVの読み込みを行います。

コード


# %%
import mysql.connector as mariadb
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import csv
import configparser

config = configparser.ConfigParser()
config.read('./define.ini')

mariadb_connection = mariadb.connect(host=config['db']['host'], user=config['db']['user'], password=config['db']['password'], database=config['db']['database'])
cursor = mariadb_connection.cursor(dictionary=True)

query = "SELECT * FROM horse_datas"
cursor.execute(query)
datas = cursor

# データをデータフレーム化する
df = pd.DataFrame(datas)

# CSVを生成する(データフレームに付与されているindex用のカラムが不要の場合はindex=Falseを引数指定する)
df = df.to_csv('./csv_dir/netkeiba.csv', index=False)

# CSVを読み込む
df = pd.read_csv('./csv_dir/netkeiba.csv')

2020年12月21日