remove_dense_vector_element.py

from pyspark.ml.linalg import Vectors

def delete_index_from_dv(dense_vector, index):
  """
  Takes PySpark dense vector, removes element at index.
  :dense_vector: PySpark dense vector to operate on.
  :index: Index of element to delete.
  :return: Dense vector with element at index removed.
  """
  temp_list = list(dense_vector)
  temp_list.pop(index)
  return Vectors.dense(temp_list)