Show parent categories and sub categories but not grand children categories

In the single post you can get the parent category like this

$category = wp_get_object_terms(get_the_ID(), 'category', array('parent'=>0));

$args = array(
 'show_count' => false,
 'hide_empty' => true,
 'title_li' => '',
 'show_option_none' => '',
 'echo' => false,
 'depth' => 1
 );


$args['child_of'] = $category[0]->term_id;

// Get the category list
$categories_list = wp_list_categories( $args );
if ( $categories_list ) {?>
  <div class="subcategory-wrapper">
    <ul class="menu">
      <?php echo $categories_list; ?>
    </ul>
  </div>
<?php } ?>