### 基于列名首字符对 DataFrame 列分组并处理行值 在 Pandas 中，可以通过字符串操作提取列名的首字符，并以此为基础对列进行分组。之后可针对每组中的行值执行特定的处理逻辑。以下是具体实现方法： #### 提取列名首字符 Pandas 的 `.str` 属性允许对列名（即索引对象）进行字符串操作。通过 `columns.str[0]` 可获取所有列名的第一个字符。 ```python import pandas as pd # 示例数据 data = { 'A_val': [1, 2, 3], 'B_score': [4, 5, 6], 'C_result': [7, 8, 9], 'A_count': [10, 11, 12], 'B_total': [13, 14, 15] } df = pd.DataFrame(data) # 提取列名首字符 first_char = df.columns.str[0] ``` 上述代码中，变量 `first_char` 存储了每个列名的第一位字符[^1]。 --- #### 按首字符分组 使用 `groupby()` 函数可以根据指定的标准对列进行分组。在这里，我们将基于提取的首字符对列进行分组。 ```python # 按首字符分组 grouped_columns = df.groupby(first_char, axis=1) ``` 此时，`grouped_columns` 是一个分组对象，表示按列名首字符分组的结果[^1]。 --- #### 行值处理 对于每个分组，可以对其内部的行值执行任意处理逻辑。例如，我们可以计算每组内各行值的最大值、最小值或其他统计量。 ```python def process_group(group): """ 对分组内的行值进行处理 """ max_values = group.max(axis=1) # 计算每行最大值 min_values = group.min(axis=1) # 计算每行最小值 return pd.concat([max_values.rename(f'{group.name}_max'), min_values.rename(f'{group.name}_min')], axis=1) # 应用处理逻辑 processed_results = pd.concat([process_group(g) for _, g in grouped_columns], axis=1) ``` 在这段代码中： - 定义了一个函数 `process_group` 来处理每个分组。 - 使用 `group.max(axis=1)` 和 `group.min(axis=1)` 分别计算每行的最大值和最小值。 - 将结果拼接成新的 DataFrame 返回[^1]。 --- #### 完整代码示例 以下是完整的代码实现： ```python import pandas as pd # 创建示例数据 data = { 'A_val': [1, 2, 3], 'B_score': [4, 5, 6], 'C_result': [7, 8, 9], 'A_count': [10, 11, 12], 'B_total': [13, 14, 15] } df = pd.DataFrame(data) # 提取列名首字符 first_char = df.columns.str[0] # 按首字符分组 grouped_columns = df.groupby(first_char, axis=1) # 定义处理逻辑 def process_group(group): max_values = group.max(axis=1).rename(f'{group.name}_max') min_values = group.min(axis=1).rename(f'{group.name}_min') return pd.concat([max_values, min_values], axis=1) # 应用处理逻辑并合并结果 processed_results = pd.concat([process_group(g) for _, g in grouped_columns], axis=1) print(processed_results) ``` 运行此代码后，将获得一个新的 DataFrame，其中包含了按列名首字符分组后的每行最大值和最小值[^1]。 --- ### 输出样例 假设输入数据为： | A_val | B_score | C_result | A_count | B_total | |-------|---------|----------|---------|---------| | 1 | 4 | 7 | 10 | 13 | | 2 | 5 | 8 | 11 | 14 | | 3 | 6 | 9 | 12 | 15 | 则输出结果可能为： | A_max | A_min | B_max | B_min | C_max | C_min | |-------|-------|-------|-------|-------|-------| | 10 | 1 | 13 | 4 | 7 | 7 | | 11 | 2 | 14 | 5 | 8 | 8 | | 12 | 3 | 15 | 6 | 9 | 9 | ---