Split CSV with bash

split_csv.sh

#!/bin/bash

# usage $1: number_files $2: filename $3: [output_path]
# example:
# $ ./split_csv.sh 2 my_dir/large_file.csv ./output_dir

if [ $# -ge 3 ]
then
  mkdir -p $3
fi

output=${3:-./}
split -nl/$1 -d --additional-suffix=.csv $2 $output/part_

# TODO support more 00000
# TODO better argument list (--help)
for file in $output/part_*
do
  if [[ $file != *art_00.csv ]]
  then
    head -n 1 $2 > tmp_file
    cat $file >> tmp_file
    mv -f tmp_file $file
  fi
done