讀取.csv文件和.xlsx文件並打印output

Question

讀取 2 個擴展名為.csv 和 .xlsx 的文件並檢查列（屬性）是否存在然后打印值。（此代碼僅適用於.csv 文件，不適用於.xlsx）。

import re
import pandas as pd


def Group_name_check(path):

 df = pd.read_csv(path)
 #name= pd.DataFrame(df)
 if "property" in df.columns:
     test=df['property'][0]
     Data=re.sub(r"[^a-zA-Z0-9]","",test)
     print("Data:",Data)
 else:
     print("Data:")

if __name__ == "__main__":
  path = "D:\53943_Sreerag_cde_one_check.xlsx"
  Group_name_check(path)

Answer 1

只需稍微修改您的代碼 - 請參閱下面的評論：

def Group_name_check(path):
    # first check if your file can be loaded as csv
    try:
        df = pd.read_csv(path)
    except:
        # if it's not csv, check if your file can be loaded as excel
        try:
            df = pd.read_excel(path)
        except:
            # in case none of it works, your file probably is not csv nor excel
            print("Error reading CSV/XLSX file.")
    #name= pd.DataFrame(df)
    if "property" in df.columns:
        test=df['property'][0]
        Data=re.sub(r"[^a-zA-Z0-9]","",test)
        print("Data:",Data)
    else:
        print("Data:")