python如何确定dataframe的最大值的位置，不用idmax

可以使用以下代码来确定dataframe的最大值的位置：

import numpy as np
import pandas as pd

# 创建一个示例dataframe
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 使用numpy的argmax函数来确定最大值的位置
max_position = np.unravel_index(np.argmax(df.values), df.shape)

# 输出最大值的位置
print(max_position)

输出结果为：

(2, 2)

这表示最大值位于第3行第3列。

相关问题

配置 Spark 通过 JDBC 连接数据库 MySQL，编程实现利用 DataFrame 插入id name gender age: 3, Mary, F, 26;4, Tom, M, 23两行数据到 MySQL 中，最后打印出 age 的最大值和 age 的总和。

下面是基于Python语言的Spark通过JDBC连接MySQL的实现代码：

# 导入相关的库
from pyspark.sql import SparkSession

# 创建SparkSession对象
spark = SparkSession.builder.appName("JdbcExample").getOrCreate()

# 配置MySQL连接参数
url = "jdbc:mysql://localhost:3306/sparktest"
table = "employee"
mode = "append"
properties = {
    "user": "username",
    "password": "password"
}

# 创建DataFrame
data = [(3, 'Mary', 'F', 26), (4, 'Tom', 'M', 23)]
df = spark.createDataFrame(data, ['id', 'name', 'gender', 'age'])

# 将DataFrame写入MySQL数据库
df.write.jdbc(url=url, table=table, mode=mode, properties=properties)

# 从MySQL数据库中读取数据到DataFrame
df = spark.read.jdbc(url=url, table=table, properties=properties)

# 计算age的最大值和总和
max_age = df.agg({"age": "max"}).collect()[0][0]
sum_age = df.agg({"age": "sum"}).collect()[0][0]

# 打印age的最大值和总和
print("Max age: {}".format(max_age))
print("Sum age: {}".format(sum_age))

需要注意的是，在代码中需要将`username`和`password`替换成自己的MySQL用户名和密码。在配置MySQL连接参数时，我们需要指定MySQL数据库的连接地址和要操作的表名，以及访问MySQL的用户名和密码。然后，我们使用`createDataFrame`方法创建DataFrame，并使用`write.jdbc`方法将DataFrame写入MySQL数据库中。最后，我们使用`read.jdbc`方法从MySQL数据库中读取数据到DataFrame中，并计算age的最大值和总和。

使用学习过的理论知识（Numpy数值分析基础、Matplotlib数据可视化基础、Pandas统计分析基础），对data.csv用户用户用电量数据进行相关处理，其中数据中有编号为1-200的200位电力用户，DATA_DATE表示时间，如2015/1/1表示2015年1月1日，KWH为用电量。请完成以下工作：1、将数据进行转置，转置后行为用户编号、列为日期、值为用户每日用电量。2、对数据中的异常数据进行识别并处理。3、统计每个用户用电数据的基本统计量，包括：最大值、最小值、均值、中位数、和、方差、偏度、峰度。4、每个用户用电数据按日差分，并求取差分结果的基本统计量，统计量同3。5、求取每个用户的5%分位数。6、每个用户按周求和并差分（一周7天，年度分开），并求取差分结果的基本统计量，统计量同三。7、统计每个用户的日用电量在其最大值0.9倍以上的次数。8、求取每个用户日为最大值/最小值的索引月份，若最大值/最小值存在于多个月份中，则输出含有最大值/最小值最多的那个月份。如1号用户的最小值为0，则看哪个月的0最多。9、求取每个用户七八月电量和与三四月电量和的比值，最大值的比值，最小值的比值，均值（日均电量）的比值。10、合并上述特征。

1、将数据进行转置，转置后行为用户编号、列为日期、值为用户每日用电量。

import pandas as pd

# 读取数据
df = pd.read_csv('data.csv')
# 转换日期格式
df['DATA_DATE'] = pd.to_datetime(df['DATA_DATE'])

# 转置数据
df_pivot = df.pivot(index='ID', columns='DATA_DATE', values='KWH')

2、对数据中的异常数据进行识别并处理。

import numpy as np

# 计算每个用户每天用电量的标准差
std = df_pivot.std(axis=1)
# 计算每个用户每天用电量的平均值
mean = df_pivot.mean(axis=1)

# 将超过平均值+3倍标准差或小于平均值-3倍标准差的数据替换为NaN
df_pivot[(df_pivot > mean.values[:, np.newaxis] + 3 * std.values[:, np.newaxis]) |
         (df_pivot < mean.values[:, np.newaxis] - 3 * std.values[:, np.newaxis])] = np.nan

3、统计每个用户用电数据的基本统计量，包括：最大值、最小值、均值、中位数、和、方差、偏度、峰度。

from scipy.stats import skew, kurtosis

# 统计每个用户用电数据的基本统计量
statistics = pd.DataFrame({
    'max': df_pivot.max(),
    'min': df_pivot.min(),
    'mean': df_pivot.mean(),
    'median': df_pivot.median(),
    'sum': df_pivot.sum(),
    'var': df_pivot.var(),
    'skew': skew(df_pivot, axis=1),
    'kurtosis': kurtosis(df_pivot, axis=1)
})

4、每个用户用电数据按日差分，并求取差分结果的基本统计量，统计量同3。

# 按日差分
df_diff = df_pivot.diff(axis=1)
# 删除第一列的NaN值
df_diff = df_diff.iloc[:, 1:]

# 统计每个用户用电数据按日差分后的基本统计量
diff_statistics = pd.DataFrame({
    'max': df_diff.max(),
    'min': df_diff.min(),
    'mean': df_diff.mean(),
    'median': df_diff.median(),
    'sum': df_diff.sum(),
    'var': df_diff.var(),
    'skew': skew(df_diff, axis=1),
    'kurtosis': kurtosis(df_diff, axis=1)
})

5、求取每个用户的5%分位数。

# 求取每个用户的5%分位数
quantile_5 = df_pivot.quantile(q=0.05, axis=1)

6、每个用户按周求和并差分（一周7天，年度分开），并求取差分结果的基本统计量，统计量同3。

# 按周求和
df_weekly = df_pivot.resample('W', axis=1).sum()
# 按年度分开
df_weekly = df_weekly.groupby(df_weekly.columns.year, axis=1)

# 差分
df_weekly_diff = df_weekly.diff(axis=1)
# 删除第一列的NaN值
df_weekly_diff = df_weekly_diff.iloc[:, 1:]

# 统计每个用户用电数据按周差分后的基本统计量
weekly_diff_statistics = pd.DataFrame({
    'max': df_weekly_diff.max(),
    'min': df_weekly_diff.min(),
    'mean': df_weekly_diff.mean(),
    'median': df_weekly_diff.median(),
    'sum': df_weekly_diff.sum(),
    'var': df_weekly_diff.var(),
    'skew': skew(df_weekly_diff, axis=1),
    'kurtosis': kurtosis(df_weekly_diff, axis=1)
})

7、统计每个用户的日用电量在其最大值0.9倍以上的次数。

# 统计每个用户的日用电量在其最大值0.9倍以上的次数
count = (df_pivot > df_pivot.max() * 0.9).sum()

8、求取每个用户日为最大值/最小值的索引月份，若最大值/最小值存在于多个月份中，则输出含有最大值/最小值最多的那个月份。如1号用户的最小值为0，则看哪个月的0最多。

# 求取每个用户日为最大值的索引月份
max_month = df_pivot.idxmax(axis=1).dt.month
# 求取每个用户日为最小值的索引月份
min_month = df_pivot.idxmin(axis=1).dt.month

# 统计每个用户日为最大值/最小值的索引月份中出现次数最多的月份
max_month_count = max_month.value_counts()
min_month_count = min_month.value_counts()

# 输出结果
print('每个用户日为最大值的索引月份：')
print(max_month[max_month == max_month_count.idxmax()].value_counts())
print('\n每个用户日为最小值的索引月份：')
print(min_month[min_month == min_month_count.idxmax()].value_counts())

9、求取每个用户七八月电量和与三四月电量和的比值，最大值的比值，最小值的比值，均值（日均电量）的比值。

# 求取每个用户七八月电量和
summer_sum = df_pivot.loc[:, df_pivot.columns.month.isin([7, 8])].sum(axis=1)
# 求取每个用户三四月电量和
spring_sum = df_pivot.loc[:, df_pivot.columns.month.isin([3, 4])].sum(axis=1)

# 求取比值
ratio = summer_sum / spring_sum

# 输出结果
print('每个用户七八月电量和与三四月电量和的比值：')
print(ratio)
print('\n最大值的比值：')
print(ratio.max())
print('\n最小值的比值：')
print(ratio.min())
print('\n均值的比值：')
print(ratio.mean())

10、合并上述特征。

# 合并特征
features = pd.concat([
    statistics,
    diff_statistics,
    pd.DataFrame({'quantile_5': quantile_5}),
    weekly_diff_statistics,
    pd.DataFrame({'count': count}),
    pd.DataFrame({'max_month': max_month, 'min_month': min_month}),
    pd.DataFrame({'ratio': ratio})
], axis=1)