【Python × マーベル映画】pandas データフレームの作り方

こんにちは！こちらの記事では、Pandas を使ってデータフレームを作成する方法を紹介します。

 環境: Python 3.9, Pandas 1.2.4

1. リストからDataFrameを作成する。

上記のマーベル映画の情報が入っているデータフレームを作成します。

import pandas as pd
from IPython.display import display

#リストのリストを作成
list_of_lists = [[1, "Iron Man", "Robert Downey, Jr."],
                 [2, "The Incredible Hulk", "Edward Norton"],
                 [3, "Iron Man 2", "Robert Downey Jr."],
                 [4, "Thor", "Chris Hemsworth"]]

# データフレームを作成
df_a = pd.DataFrame(list_of_lists, columns=["FilmID", "FilmName","MainActor"])

display(df_a) 

2. 複数のリストからDataFrameを作成する。

上記のマーベル映画の情報が入っているデータフレームを作成します。

# Import library
import pandas as pd
from IPython.display import display

# 複数のリストを作成
Film_ID = [1, 2, 3, 4]
Film_name = ["Iron Man", "The Incredible Hulk", "Iron Man 2","Thor"]
Main_actor = ["Robert Downey Jr.", "Edward Norton", "Robert Downey Jr.", "Chris Hemsworth"]

# データフレームを作成
df_b = pd.DataFrame(list(zip(Film_ID, Film_name, Main_actor)),
                  columns=["FilmID", "FilmName","MainActor"])

display(df_b)

3. 辞書のリストからDataFrameを作成する。

上記のマーベル映画の情報が入っているデータフレームを作成します。

import pandas as pd
from IPython.display import display

# 辞書のリストを作成
dictionary_of_lists = {
    "FilmID": [1, 2, 3, 4],
    "FilmName": ["Iron Man", "The Incredible Hulk", "Iron Man 2","Thor"], 
    "MainActor": ["Robert Downey Jr.", "Edward Norton", "Robert Downey Jr.", "Chris Hemsworth"]}

# データフレームを作成
df_c = pd.DataFrame(dictionary_of_lists)
display(df_c)

4. インデックスをつけて、データフレームを作成します。

上記のマーベル映画の情報が入っているデータフレームを作成します。

import pandas as pd
from IPython.display import display

# インデックスをつけて、データフレームを作成
df_d = pd.DataFrame(columns=["FilmID", "FilmName","ActorName"], index=['a', 'b', 'c','d'])
df_d.loc['a'] = [1, "Iron Man","Robert Downey Jr."] #Rober"s name should be counted as three pieces
df_d.loc['b'] = [2, "The Incredible Hulk", "Mark Ruffalo"]
df_d.loc['c'] = [3, "Iron Man 2", "Robert Downey Jr."]
df_d.loc['d'] = [4, "Thor", "Chris Hemsworth"]

display(df_d)

今回は、以上となります。ありがとうございました。