与熊猫不同的名字序列
我有数据框
used_at common users pair of websites
0 2014 1364 avito.ru and e1.ru
1 2014 1716 avito.ru and drom.ru
2 2014 1602 avito.ru and auto.ru
3 2014 299 avito.ru and avtomarket.ru
4 2014 579 avito.ru and am.ru
5 2014 602 avito.ru and irr.ru/cars
6 2014 424 avito.ru and cars.mail.ru/sale
7 2014 634 e1.ru and drom.ru
8 2014 475 e1.ru and auto.ru
9 2014 139 e1.ru and avtomarket.ru
10 2014 224 e1.ru and am.ru
11 2014 235 e1.ru and irr.ru/cars
12 2014 154 e1.ru and cars.mail.ru/sale
13 2014 874 drom.ru and auto.ru
14 2014 247 drom.ru and avtomarket.ru
15 2014 394 drom.ru and am.ru
....
当我写graph_by_common_users = common_users.pivot(index='pair of websites', columns='used_at', values='common users')
时
我得到了
used_at 2014 2015
pair of websites
am.ru and cars.mail.ru/sale 166.0 NaN
am.ru and irr.ru/cars 223.0 NaN
auto.ru and am.ru 408.0 224.0
auto.ru and avtomarket.ru 243.0 162.0
auto.ru and cars.mail.ru/sale 330.0 195.0
auto.ru and drom.ru NaN 799.0
auto.ru and irr.ru/cars 409.0 288.0
avito.ru and am.ru 579.0 262.0
....
我有NaN，因为有些序列不同。例如
我有2014我有am.ru and cars.mail.ru/sale但2015我有cars.mail.ru/sale and am.ru。我怎么能改变呢？
添加我的代码
import pandas as pd
import itertools
import matplotlib.pyplot as plt
df = pd.read_csv("avito_trend.csv", parse_dates=[2])
def f(df):
dfs = []
for x in [list(x) for x in itertools.combinations(df['address'].unique(), 2)]:
c1 = df.loc[df['address'].isin([x[0]]), 'ID']
c2 = df.loc[df['address'].isin([x[1]]), 'ID']
c = pd.Series(list(set(c1).intersection(set(c2))))
dfs.append(pd.DataFrame({'common users':len(c), 'pair of websites':' and '.join(x)}, index=[0]))
return pd.concat(dfs)
common_users = df.groupby([df['used_at'].dt.year]).apply(f).reset_index(drop=True, level=1).reset_index()
print common_users
graph_by_common_users = common_users.pivot(index='pair of websites', columns='used_at', values='common users')
print graph_by_common_users
2 个答案:
答案 0 :(得分：2)
也许在转动之前，尝试拆分" and "然后排序，以便每列的顺序相同：
df['pair of websites'] = df['pair of websites'].str.split(' and ')
df['pair of websites'] = df['pair of websites'].apply(lambda x: frozenset(sorted(x)))
似乎只要在每个条目的“和”部分中都有相同数量的空白，这应该可以正常工作。如果没有，您可能还必须使用str.strip()。
答案 1 :(得分：1)
经过测试后，我添加了倒置组合c_invert，因为在pivot之后缺少某些值。现在所有combination和pivot都能很好地运作：
df = pd.read_csv("avito_trend.csv",
parse_dates=[2])
def f(df):
dfs = []
for x in [list(x) for x in itertools.combinations(df['address'].unique(), 2)]:
c1 = df.loc[df['address'].isin([x[0]]), 'ID']
c2 = df.loc[df['address'].isin([x[1]]), 'ID']
c = pd.Series(list(set(c1).intersection(set(c2))))
#add inverted intersection c2 vs c1
c_invert = pd.Series(list(set(c2).intersection(set(c1))))
dfs.append(pd.DataFrame({'common users':len(c), 'pair of websites':' and '.join(x)}, index=[0]))
#swap values in x
x[1],x[0] = x[0],x[1]
dfs.append(pd.DataFrame({'common users':len(c_invert), 'pair of websites':' and '.join(x)}, index=[0]))
return pd.concat(dfs)
common_users = df.groupby([df['used_at'].dt.year]).apply(f).reset_index(drop=True, level=1).reset_index()
print common_users.pivot(index='pair of websites', columns='used_at', values='common users')
used_at 2014 2015
pair of websites
am.ru and auto.ru 408 224
am.ru and avito.ru 579 262
am.ru and avtomarket.ru 133 72
am.ru and cars.mail.ru/sale 166 73
am.ru and drom.ru 394 187
am.ru and e1.ru 224 99
am.ru and irr.ru/cars 223 102
auto.ru and am.ru 408 224
auto.ru and avito.ru 1602 1473
auto.ru and avtomarket.ru 243 162
auto.ru and cars.mail.ru/sale 330 195
auto.ru and drom.ru 874 799
auto.ru and e1.ru 475 451
auto.ru and irr.ru/cars 409 288
avito.ru and am.ru 579 262
avito.ru and auto.ru 1602 1473
avito.ru and avtomarket.ru 299 205
avito.ru and cars.mail.ru/sale 424 256
avito.ru and drom.ru 1716 1491
avito.ru and e1.ru 1364 1153
avito.ru and irr.ru/cars 602 403
avtomarket.ru and am.ru 133 72
avtomarket.ru and auto.ru 243 162
avtomarket.ru and avito.ru 299 205
avtomarket.ru and cars.mail.ru/sale 105 48
avtomarket.ru and drom.ru 247 175
avtomarket.ru and e1.ru 139 105
avtomarket.ru and irr.ru/cars 139 73
cars.mail.ru/sale and am.ru 166 73
cars.mail.ru/sale and auto.ru 330 195
cars.mail.ru/sale and avito.ru 424 256
cars.mail.ru/sale and avtomarket.ru 105 48
cars.mail.ru/sale and drom.ru 292 189
cars.mail.ru/sale and e1.ru 154 105
cars.mail.ru/sale and irr.ru/cars 197 94
drom.ru and am.ru 394 187
drom.ru and auto.ru 874 799
drom.ru and avito.ru 1716 1491
drom.ru and avtomarket.ru 247 175
drom.ru and cars.mail.ru/sale 292 189
drom.ru and e1.ru 634 539
drom.ru and irr.ru/cars 423 277
e1.ru and am.ru 224 99
e1.ru and auto.ru 475 451
e1.ru and avito.ru 1364 1153
e1.ru and avtomarket.ru 139 105
e1.ru and cars.mail.ru/sale 154 105
e1.ru and drom.ru 634 539
e1.ru and irr.ru/cars 235 148
irr.ru/cars and am.ru 223 102
irr.ru/cars and auto.ru 409 288
irr.ru/cars and avito.ru 602 403
irr.ru/cars and avtomarket.ru 139 73
irr.ru/cars and cars.mail.ru/sale 197 94
irr.ru/cars and drom.ru 423 277
irr.ru/cars and e1.ru 235 148
如果您需要图表：
graph_by_common_users = common_users.pivot(index='pair of websites', columns='used_at', values='common users')
#sort by column 2014
graph_by_common_users = graph_by_common_users.sort_values(2014, ascending=False)
ax = graph_by_common_users.plot(kind='barh', width=0.5, figsize=(10,20))
[label.set_rotation(25) for label in ax.get_xticklabels()]
rects = ax.patches
labels = [int(round(graph_by_common_users.loc[i, y])) for y in graph_by_common_users.columns.tolist() for i in graph_by_common_users.index]
for rect, label in zip(rects, labels):
height = rect.get_height()
ax.text(rect.get_width() + 3, rect.get_y() + rect.get_height(), label, fontsize=8)
相关问题
最新问题
- 我写了这段代码，但我无法理解我的错误
- 我无法从一个代码实例的列表中删除 None 值，但我可以在另一个实例中。为什么它适用于一个细分市场而不适用于另一个细分市场？
- 是否有可能使 loadstring 不可能等于打印？卢阿
- java中的random.expovariate()
- Appscript 通过会议在 Google 日历中发送电子邮件和创建活动
- 为什么我的 Onclick 箭头功能在 React 中不起作用？
- 在此代码中是否有使用“this”的替代方法？
- 在 SQL Server 和 PostgreSQL 上查询，我如何从第一个表获得第二个表的可视化
- 每千个数字得到
- 更新了城市边界 KML 文件的来源？