既存のCSVファイル2列めにデータを追加したい

Question

こんにちは! 自然言語処理をかけてcsvに登場回数順に出力したロシア語をgoogletransにかけ、列の3番めに日本語訳を追記したいのですが、うまく列を指定できません。

どうしたら2列めに日本語訳が入るようになるでしょうかorz

開発環境

• Linux Mint19 ”Tara” Cinammon
• Pycharm
• Python3.6.6

現在の出力結果と望んでいる結果

現在の出力結果

望んでいる結果

該当するコード

# -*- coding: utf-8 -*-
from nltk import FreqDist
import pymorphy2
import csv
import re

import del_words 
from csvScraping import show_word

downloadtxt = open('26and1.txt', 'r', encoding = 'utf-8-sig')
texts = downloadtxt.read()
texts = re.sub('[.,?!()\'—]', ' ', texts)
texts = texts.split()

analyzer = pymorphy2.MorphAnalyzer()

lemma_list = []
for text in texts:
    lemma = analyzer.parse(text)[0].normal_form
    lemma_list.append(lemma)

lemma_list = list(filter(lambda d: d not in (str(del_words.del_w)), lemma_list))
freqdist = FreqDist(lemma_list)
freqdist_list = freqdist.most_common(10) # 900
print(freqdist_list)
with open('26and1.csv', 'w', encoding = 'utf-8-sig') as file:
    writer = csv.writer(file, lineterminator = '\n')
    writer.writerow(['Ру','Ном','Яп'])
    for n in range(0, len(freqdist_list)):
        writer.writerows([freqdist_list[n]])
pass

show_word() # csvScraping.py

# csvScraping.py
import pandas as pd
import csv
from googletrans import Translator

def show_word():
    csv_input = pd.read_csv(filepath_or_buffer= r'26and1.csv', sep=",")

    with open('26and1.csv', 'r', encoding='utf-8-sig') as file:
        for i in range(0, 10):
            ru = csv_input.iloc[i, 0]
            translator = Translator()
            ru_ja = translator.translate(ru, 'ja')
            ja = ru_ja.text
            print(ja)

print(ja)の部分をcsvのファイル操作に差し替えればよいのかとは思うのですが、「n列目を指定してそこにデータを追記」という方法が見つけられませんorz
よろしくお願いします。

できましたー(≧∇≦)b

csvScraping.pyのみの変更です。
csvの内容['項目名'][index番号] で指定<3

# csvScraping.py
import pandas as pd
from googletrans import Translator

def show_word():
    csv_input = pd.read_csv('26and1.csv', sep=",")
    csv_input['Яп'] = ''

    for index, row in csv_input.iterrows():
        ru = csv_input.iloc[index, 0]
        translator = Translator()
        ru_ja = translator.translate(ru, 'ja')
        ja = ru_ja.text
        csv_input['Яп'][index] = ja

    csv_input.to_csv('26and1.csv', index=False)

Answer 1

pandas を使う例と標準モジュールの csv を使う例を書きました。

pandas を使わない場合のサンプルコード

import csv

data = [
    ['apple', 1],
    ['pen', 2],
    ['pineapple', 3],
]

# 列を追加
for row in data:
    row.append('new_data')  # 参照なので、data に反映される。
print(data)
# [['apple', 1, 'new_data'], ['pen', 2, 'new_data'], ['pineapple', 3, 'new_data']]

with open('test.csv', 'w') as f:
    writer = csv.writer(f)
    writer.writerows(data)

print(open('test.csv').read())
# apple,1,new_data
# pen,2,new_data
# pineapple,3,new_data

pandas を使う場合のサンプルコード

import pandas as pd

data = pd.read_csv('test.csv', sep=",")
# 予め列を挿入しておく必要がある。
data['new_column'] = ''

# 列に値を代入
for index, row in data.iterrows():
    data['new_column'][index] = 'hoge'

data.to_csv('test.csv', index=False)

print(open('test.csv').read())
# name,count,new_column
# apple,1,hoge
# pen,2,hoge
# pineapple,3,hoge

Comments

[Yukiya025]

tiitoiさん、こんばんは！ ありがとうございます、できましたー(≧∇≦)b
pandasを使う方法で日本語訳が2列めに追記できました<3