import pandas as pd
import csv
from io import StringIO

def csv_to_dataframe(csv_data=None, file_path=None, delimiter='\t'):
    """
    將CSV數(shù)據(jù)轉(zhuǎn)換為DataFrame
    
    參數(shù):
    csv_data: CSV格式的字符串?dāng)?shù)據(jù)
    file_path: CSV文件路徑
    delimiter: 分隔符，默認(rèn)為制表符
    
    返回:
    pandas DataFrame對象
    """
    if csv_data:
        # 從字符串讀取數(shù)據(jù)
        df = pd.read_csv(StringIO(csv_data), delimiter=delimiter)
    elif file_path:
        # 從文件讀取數(shù)據(jù)
        df = pd.read_csv(file_path, delimiter=delimiter)
    else:
        raise ValueError("必須提供csv_data或file_path參數(shù)")
    
    return df

def filter_dataframe(df, **kwargs):
    """
    根據(jù)多列條件篩選DataFrame
    
    參數(shù):
    df: 要篩選的DataFrame
    **kwargs: 篩選條件，格式為 列名=值 或 列名=[值1, 值2, ...]
    
    返回:
    篩選后的DataFrame
    """
    # 初始化篩選條件
    condition = pd.Series([True] * len(df))
    
    # 處理每個篩選條件
    for column, values in kwargs.items():
        if column not in df.columns:
            raise ValueError(f"列名 '{column}' 不存在于DataFrame中")
        
        # 如果值是列表，使用isin方法
        if isinstance(values, list):
            condition = condition & df[column].isin(values)
        # 如果是單個值，使用等號比較
        else:
            condition = condition & (df[column] == values)
    
    # 應(yīng)用篩選條件
    filtered_df = df[condition]
    
    return filtered_df

# 示例使用
if __name__ == "__main__":
    # 示例數(shù)據(jù)
    csv_data = """!+&	idx	Market	CURR	PM	TERMANT
*	1	JP	USD	1	0
*	1	CHINA	EUD	0	0
*	1	USA	USD	1	0
*	2	JP	JPY	1	0
*	3	USA	CNY	1	1
*	4	CHINA	CNY	0	0
*	5	JP	USD	1	1
*	6	JP	JPY	0	0"""
    
    # 使用第一個函數(shù)將CSV數(shù)據(jù)轉(zhuǎn)換為DataFrame
    df = csv_to_dataframe(csv_data=csv_data, delimiter='\t')
    print("原始數(shù)據(jù):")
    print(df)
    print("\n" + "="*50 + "\n")
    
    # 示例1: 篩選出Market為JP且CURR為JPY的所有行
    result1 = filter_dataframe(df, Market='JP', CURR='JPY')
    print("示例1 - Market為JP且CURR為JPY:")
    print(result1)
    print("\n" + "="*50 + "\n")
    
    # 示例2: 篩選出Market為日本和美國，CURR為USD和CNY
    result2 = filter_dataframe(df, Market=['JP', 'USA'], CURR=['USD', 'CNY'])
    print("示例2 - Market為JP或USA，CURR為USD或CNY:")
    print(result2)
    print("\n" + "="*50 + "\n")
    
    # 示例3: 篩選出Market為JP，CURR為USD、CNY和EUD
    result3 = filter_dataframe(df, Market='JP', CURR=['USD', 'CNY', 'EUD'])
    print("示例3 - Market為JP，CURR為USD、CNY或EUD:")
    print(result3)

import pandas as pd
import csv

def read_csv_to_dataframe(file_path, delimiter='\t', encoding='utf-8'):
    """
    從指定文件路徑讀取CSV數(shù)據(jù)并轉(zhuǎn)換為DataFrame
    
    參數(shù):
    file_path: CSV文件的路徑
    delimiter: 分隔符，默認(rèn)為制表符
    encoding: 文件編碼，默認(rèn)為utf-8
    
    返回:
    pandas DataFrame對象
    """
    try:
        # 讀取CSV文件
        df = pd.read_csv(file_path, delimiter=delimiter, encoding=encoding)
        print(f"成功讀取文件: {file_path}")
        print(f"數(shù)據(jù)形狀: {df.shape}")
        return df
    except FileNotFoundError:
        print(f"錯誤: 文件 '{file_path}' 不存在")
        return None
    except Exception as e:
        print(f"讀取文件時出錯: {str(e)}")
        return None

def filter_dataframe(df, **kwargs):
    """
    根據(jù)多列條件篩選DataFrame
    
    參數(shù):
    df: 要篩選的DataFrame
    **kwargs: 篩選條件，格式為 列名=值 或 列名=[值1, 值2, ...]
    
    返回:
    篩選后的DataFrame
    """
    if df is None or df.empty:
        print("錯誤: DataFrame為空或未提供有效數(shù)據(jù)")
        return None
    
    # 初始化篩選條件
    condition = pd.Series([True] * len(df))
    
    # 處理每個篩選條件
    for column, values in kwargs.items():
        if column not in df.columns:
            print(f"警告: 列名 '{column}' 不存在于DataFrame中，已跳過此條件")
            continue
        
        # 如果值是列表，使用isin方法
        if isinstance(values, list):
            condition = condition & df[column].isin(values)
        # 如果是單個值，使用等號比較
        else:
            condition = condition & (df[column] == values)
    
    # 應(yīng)用篩選條件
    filtered_df = df[condition]
    
    print(f"篩選后數(shù)據(jù)形狀: {filtered_df.shape}")
    return filtered_df

# 示例使用
if __name__ == "__main__":
    # 示例文件路徑
    file_path = "data.csv"
    
    # 使用第一個函數(shù)讀取CSV文件
    df = read_csv_to_dataframe(file_path)
    
    if df is not None:
        print("\n原始數(shù)據(jù)前5行:")
        print(df.head())
        print("\n" + "="*50 + "\n")
        
        # 示例1: 篩選出Market為JP且CURR為JPY的所有行
        result1 = filter_dataframe(df, Market='JP', CURR='JPY')
        print("示例1 - Market為JP且CURR為JPY:")
        print(result1)
        print("\n" + "="*50 + "\n")
        
        # 示例2: 篩選出Market為日本和美國，CURR為USD和CNY
        result2 = filter_dataframe(df, Market=['JP', 'USA'], CURR=['USD', 'CNY'])
        print("示例2 - Market為JP或USA，CURR為USD或CNY:")
        print(result2)
        print("\n" + "="*50 + "\n")
        
        # 示例3: 篩選出Market為JP，CURR為USD、CNY和EUD
        result3 = filter_dataframe(df, Market='JP', CURR=['USD', 'CNY', 'EUD'])
        print("示例3 - Market為JP，CURR為USD、CNY或EUD:")
        print(result3)
        
        # 可選: 將篩選結(jié)果保存到新文件
        if result1 is not None and not result1.empty:
            result1.to_csv("filtered_result.csv", index=False, sep='\t')
            print("篩選結(jié)果已保存到 filtered_result.csv")

# 讀取CSV文件
df = read_csv_to_dataframe("your_data.csv", delimiter='\t')

# 篩選Market為JP且CURR為JPY的行
result = filter_dataframe(df, Market='JP', CURR='JPY')

# 篩選Market為JP或USA，且CURR為USD或CNY的行
result = filter_dataframe(df, Market=['JP', 'USA'], CURR=['USD', 'CNY'])

# 篩選Market為JP，且CURR為USD、CNY或EUD的行
result = filter_dataframe(df, Market='JP', CURR=['USD', 'CNY', 'EUD'])