【Snowflake】Pythonシートからテーブルにレコード追加｜Pandasデータフレームの取込方法も言及

// データベース作成
CREATE OR REPLACE DATABASE TEST_DB;

// スキーマ作成
CREATE OR REPLACE SCHEMA TEST_DB.PYTHON_DATA;

// テーブル作成
CREATE OR REPLACE TABLE TEST_DB.PYTHON_DATA.IRIS(
    Sepal_Length FLOAT,
    Sepal_Width FLOAT,
    Petal_Length FLOAT,
    Petal_Width FLOAT,
    Species INTEGER
    );

# Snowflakeのライブラリ
import snowflake.snowpark as snowpark
from snowflake.snowpark.functions import col
from snowflake.snowpark.types import IntegerType, FloatType, StringType, StructType, StructField

# 外部ライブラリ
import pandas as pd
from sklearn import datasets


# メイン関数
def main(session: snowpark.Session): 

    # =======================================================================
    # Scikit-learnのデータセット読込
    # =======================================================================
    
    # サンプルデータ読込
    iris = datasets.load_iris()

    # 説明変数
    X = pd.DataFrame(iris.data, columns=["Sepal Length", "Sepal Width", "Petal Length", "Petal Width"],)
    
    # 目的変数
    Y = pd.DataFrame(iris.target, columns=["Species"])
    
    # 上記結合
    df_pandas = pd.concat([X,Y],axis=1)

    # =======================================================================
    # Snowflakeのデータフレームに変換
    # =======================================================================
    
    # スキーマ作成
    schema   = StructType([StructField("Sepal_Length", FloatType()), 
                           StructField("Sepal_Width",  FloatType()),
                           StructField("Petal_Length", FloatType()),
                           StructField("Petal_Width",  FloatType()),
                           StructField("Species",      IntegerType()),
                                       ])
    
    # データフレーム（Snowflake）
    df_snowflake = session.create_dataframe(df_pandas.values.tolist()[0:100],   # 説明変数
                                            schema,                             # フィールド名
                                           )


    # =======================================================================
    # データベースにレコード追加
    # =======================================================================

    df_snowflake.write.mode("overwrite").save_as_table("TEST_DB.PYTHON_DATA.IRIS")
    
    return df_snowflake

# SEPAL_LENGTH	SEPAL_WIDTH	PETAL_LENGTH	PETAL_WIDTH	  SPECIES
# 5.1	                  3.5	         1.4	        0.2	        0
# 4.9	                  3 	         1.4	        0.2	        0
# 4.7	                  3.2	         1.3	        0.2	        1
# 4.6	                  3.1	         1.5	        0.2	        1

// テーブル参照
SELECT * FROM TEST_DB.PYTHON_DATA.IRIS;

# SEPAL_LENGTH	SEPAL_WIDTH	PETAL_LENGTH	PETAL_WIDTH	  SPECIES
# 5.1	                  3.5	         1.4	        0.2	        0
# 4.9	                  3 	         1.4	        0.2	        0
# 4.7	                  3.2	         1.3	        0.2	        1
# 4.6	                  3.1	         1.5	        0.2	        1