我有一个简短的问题:有没有办法将散点图中的点显示为加号和减号?例如,此代码生成带有蓝色和红色点的散点图:
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
df = pd.DataFrame({'x': np.arange(25), 'y': np.random.normal(0,2500,25)})
fig, ax = plt.subplots()
ax.scatter(df.x, df.y, c=np.sign(df.y), cmap="bwr")
plt.show()
Run Code Online (Sandbox Code Playgroud)
我希望将蓝点作为蓝色加号,将红点作为红色减号。是否可以?提前致谢。
这应该很简单,但我仍然没有找到方法。我必须计算一个新列,其值为列 col1 和 col2 的最大值。因此,如果 col1 为 2,col2 为 4,则 new_col 应该为 4。依此类推。它位于 Pyspark 数据框中。我尝试过df=df.withColumn("new_col",max("col1","col2"))
,但收到错误“_() 需要 1 个位置参数,但给出了 2 个”。那么正确的方法是什么呢?提前致谢。