DataFrame 保留前几列和后几列

import pandas as pd

data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35],
    'Salary': [50000, 60000, 45000],
    'Location': ['NY', 'CA', 'TX'],
    'Department': ['HR', 'IT', 'Finance']
}

df = pd.DataFrame(data)
print("Original DataFrame:")
print(df)

# 保留前两列和最后两列
columns_to_keep = df.columns[[0, 1]].tolist() + df.columns[-2:].tolist()
new_df = df[columns_to_keep]
print("\nDataFrame with selected columns:")
print(new_df)