使用python函数(Def)返回所需的行/列

import pandas as pd

# 返回指定的行
def get_rows(dataframe, rows):
    return dataframe.iloc[rows]

# 返回指定的列
def get_columns(dataframe, columns):
    return dataframe[columns]

# 示例数据
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'London', 'Paris', 'Tokyo']}
df = pd.DataFrame(data)

# 返回第1行和第3行
selected_rows = get_rows(df, [0, 2])
print(selected_rows)

# 返回'Name'和'City'列
selected_columns = get_columns(df, ['Name', 'City'])
print(selected_columns)

      Name  Age     City
0    Alice   25  New York
2  Charlie   35    Paris

      Name       City
0    Alice  New York
1      Bob     London
2  Charlie      Paris
3    David      Tokyo