2019-08-01から1ヶ月間の記事一覧

条件指定付きの順列作成

# 組合せチェック def data_check(head): data = [1, 2, 1] n_head = len(head) n_data = len(data) n = n_head if n_head < n_data else n_data if head[:n] == data[:n]: return True else: return False def perm(head, rest): # 条件確認 if not data_ch…

同じものを含む順列

more-itertools.readthedocs.io pip install more_itertoos import more_itertools data = [1, 1, 1, 1, 2, 2, 2, 3, 3, 4] result = sorted(more_itertools.distinct_permutations(data)) print(result)

Pandasでグループ別にファイル保存

import pandas as pd df = pd.read_csv("2018.tsv", sep='\t') grouped_df = df.groupby("id") # print(grouped_df.groups) for id in grouped_df.groups: d = grouped_df.get_group(id) # ファイル名作成 filename = str(id) + ".tsv" # ファイル保存 d.to_…