データの持ち方を変更したい

Question

前提・実現したいこと

スプレッドシートまたはCSVにて下記図Aのように１行で格納している複数の列データを、スプレッドシートまたはCSVにて図Bのように複数行に分解して出力したいです。会社によって単価・数量が変化します。１しかない場合もあるし、３まである場合もあり、会社によりまちまちです。

使用する言語はPythonまたはGASにて行いたいです。

◾️図A
|会社名|単価１|数量１|単価２|数量２|単価３|数量３|
|:--|:--:|--:|
|株式会社A|380|50|3000|2|5000|10|
|株式会社B|50|100|8000|1|||

◾️図B

会社名	単価	数量
株式会社A	380	50
株式会社A	3000	2
株式会社A	5000	10
株式会社B	50	100
株式会社B	8000	1

どのように記述していいかアイデアが出てこないのでご教授宜しくお願い致します。

補足情報（FW/ツールのバージョンなど）

mac OS High Sierra, Python3

Answer 1

function q220924() {
    const csvId = '****';
    const dat = Utilities.parseCsv(DriveApp.getFileById(csvId).getBlob().getDataAsString(), ',').slice(1).reduce(function (a, c) {
        return a.concat(c);
    }, []);
    SpreadsheetApp.getActive().insertSheet().getRange(1, 1, dat.length + 1, 3).setValues([["会社", "単価", "数量"]].concat(dat));
}

function chunk(row) {
    const company = row[0];
    return row.slice(1).filter(function (e) {
        return e !== "";
    }).reduce(function (a, _, i, array) {
        if (!(i % 2)) {
            a.push([company, array[i], array[i + 1]]);
            ;
        }
        return a;
    }, [])
}

Answer 2

Pythonで実施する場合はこんな感じできると思います。

import pandas as pd

# NaNが混在している列が浮動小数点になるのを回避するため「dtype=str」をつけて文字列として扱う
df = pd.read_csv('input.csv', dtype=str)

print(df)
"""
     会社名  単価１  数量１   単価２ 数量２   単価３  数量３
0  株式会社A  380   50  3000   2  5000   10
1  株式会社B   50  100  8000   1   NaN  NaN
"""

columns = [ ['単価１', '数量１'], ['単価２', '数量２'], ['単価３', '数量３'] ]

# 列名を変えながら結合
df = pd.concat([ df.loc[:,['会社名', x[0], x[1]]].rename(columns={ x[0] : '単価', x[1]: '数量'}) for x in columns ])

# NaNがある行をdrop＆ソート&index振り直し
df = df.dropna(how='all',subset=['単価', '数量']).sort_values(by=['会社名']).reset_index(drop=True)

print(df)
"""
     会社名    単価   数量
0  株式会社A   380   50
1  株式会社A  3000    2
2  株式会社A  5000   10
3  株式会社B    50  100
4  株式会社B  8000    1
"""

df.to_csv('output.csv', index=False)

Comments

[kumer1]

ありがとうございました。
実現できました。

Answer 3

はいでは、プログラムの基本的なことを教わってなく、完全独学で勉強してる私がとりあえず先に回答しておきます。

function myFunction() {

  var sh = "シート化します"
  var arr = sh.getRange("タイトル除いた最初～最後まで").getValues()
  var arr2 = []
  var record = []
  var kaisya = []
 
  
  for (var a = 0; a < arr.length;a++)
  {
    kaisya = arr[a][0]//会社名だけ別においとく。2行目になったら二行目の会社名をとる
    for (var b = 0; b < arr[a].length; b++)
      {
        
        var num = arr[a][b*2+1]//単価情報
        var num2 = arr[a][b*2+2]//数量情報
        if(num == undefined || num == ""){break;}
        record.push(kaisya,num,num2)//図Bのようにrecord = [会社名,単価,数量]にする
        arr2.push(record)//そのレコードをプッシュする
        record = []//レコードをまっさらにする。
      }  
       
  }
  
  sh.getRange("開始Row","開始Col",arr2.length,arr2[0].lenth).setValues(arr2)
  
}

実際にシート化したりレンジ充てたりしてないのでちゃんと動くかは不明ですので発想だけを伝えますと、

forの中にfor文をいれて、
1つめのfor文には会社名をコピーして、2つめのfor文で2つづつ取っていって、それをレコード化しています。

ごり押しな考え方なので、単価、数量以外に何かはいったりしたら正しい結果になりません。

もう少し待てばもっと素晴らしい回答が頂けると思うので、とりあえず先にこの発想で組んでみて下さい。

Comments

[kumer1]

ありがとうございました。実現できました。

Answer 4

c.py

import csv
import pprint

with open('data.csv') as f:
    reader = csv.DictReader(f)
    rows = [row for row in reader]

# pprint.pprint(rows)
rows2 = []
for row in rows:
    camp = row["会社名"]
    for key in row.keys():
        if key.startswith("単価"):
            key_num = key.replace("単価", "数量", 1)
            if row[key] and row[key_num]:
                rows2.append({"会社名": camp, "単価": row[key], "数量": row[key_num]})

# pprint.pprint(rows2)

with open('data2.csv', 'w') as f:
    writer = csv.DictWriter(f, ["会社名", "単価", "数量"])
    writer.writeheader()
    for row in rows2:
        writer.writerow(row)

実行例：

参考情報

• PythonでCSVファイルを読み込み・書き込み（入力・出力）

https://note.nkmk.me/python-csv-reader-writer/

Comments

[kumer1]

ありがとうございました。応用できやすかったのでベストアンサーにさせていただきました。