Pythonのpandasのサンプル

自分勉強記録

基本動作

基本

import pandas as pd

# リストからデータフレームを作成する
data = {
    '名前': ['Alice', 'Bob', 'Charlie'],
    '年齢': [25, 30, 35],
    '性別': ['女性', '男性', 'その他']
}
df = pd.DataFrame(data)

# データフレームを表示する
print(df)

# データフレームの列を取得する
names = df['名前']
ages = df['年齢']

# 列を追加する
df['都市'] = ['東京', '大阪', '札幌']

# 条件に合うデータをフィルタリングする
filtered_df = df[df['年齢'] > 30]

# データフレームをCSVファイルに保存する
df.to_csv('output.csv', index=False)

列名取得

import pandas as pd

# データフレームを作成する
data = {
    '名前': ['Alice', 'Bob', 'Charlie'],
    '年齢': [25, 30, 35],
    '性別': ['女性', '男性', 'その他']
}
df = pd.DataFrame(data)

# 列名を取得する
column_names = df.columns

# 列名を表示する
for column in column_names:
    print(column)

列名変更

import pandas as pd

# データフレームを作成する
data = {
    '名前': ['Alice', 'Bob', 'Charlie'],
    '年齢': [25, 30, 35],
    '性別': ['女性', '男性', 'その他']
}
df = pd.DataFrame(data)

# 列名を変更する
df = df.rename(columns={'名前': 'Name', '年齢': 'Age', '性別': 'Gender'})

# 先頭の2つの列名を変更する
df.columns = df.columns.astype(str)
df.columns.values[:2] = ['New_Name1', 'New_Name2']

# 変更後のデータフレームを表示する
print(df)

指定フォルダ配下のtxtファイルを一覧

import os

def list_files(directory):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(".txt"):
                print(os.path.join(root, file))

# 指定フォルダのパスを指定
folder_path = "/path/to/folder"

# フォルダ内の.txtファイルを一覧表示
list_files(folder_path)