如何使用來自 pandas dataframe 的 sql 文件創建 txt 文件夾列表

Question

我想使用基於 dataframe 列的 sql 文件創建 txt 文件夾列表。

例子：

我需要創建名為我的 main_folder 列的 txt 文件夾列表，其中包含名為 my sql_file_x 的 sql 查詢的 ZAC5C74B64B4B4B8352EF2F181AFFB5AC2AZ 文件，其中 sql 查詢來自列 sql_query_x 和 sql 查詢，查詢來自列 mysql_query_sql 文件。

Answer 1

您可以使用 pandas apply 來進行文件操作。 這是一個例子：

x='''main_folder,sql_file_x,sql_file_y,sql_query_x,sql_query_y
"a1111","a1111.sql","data111.sql","select 111","select 'dd',111"
"a2222","a2222.sql","data222.sql","select 222","select 'dd',222"
"a3333","a3333.sql","data333.sql","select 333","select 'dd',333"
'''
import pandas as pd, io, os
df=pd.read_csv(io.StringIO(x))

def createFilesFolders(row):
    os.makedirs(row.main_folder,exist_ok=True)
    with open(os.path.join(row.main_folder,row.sql_file_x),"w") as f1:
        f1.write(row.sql_query_x)
    with open(os.path.join(row.main_folder,row.sql_file_y),"w") as f2:
        f2.write(row.sql_query_y)

df.apply(createFilesFolders, axis=1)

Output：

.
├── a1111
│   ├── a1111.sql
│   └── data111.sql
├── a2222
│   ├── a2222.sql
│   └── data222.sql
├── a3333
│   ├── a3333.sql
│   └── data333.sql
└── solution.ipynb